flax
Fix nnx.Dropout crash when broadcast_dims includes a sharded axis
#5528
Open
Overview
Commits
1
Changes
View On GitHub

Fix nnx.Dropout crash when broadcast_dims includes a sharded axis #5528

sohumt123 wants to merge 1 commit into google:main from sohumt123:fix-dropout-broadcast-dims-sharding
sohumt123
sohumt123 Fix nnx.Dropout crash when broadcast_dims includes a sharded axis
a0ce6ec4
google-cla

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum