flax
support explicit sharding in eager sharding
#5070
Merged

support explicit sharding in eager sharding #5070

cgarciae
jburnim
jburnim approved these changes on 2025-11-04
jburnim
jburnim commented on 2025-11-04
cgarciae support explicit sharding in eager sharding
c05d004d
cgarciae cgarciae force pushed from dd430017 to c05d004d 162 days ago
cgarciae cgarciae added pull ready
copybara-service copybara-service merged de47d9b7 into main 156 days ago
copybara-service copybara-service deleted the variable-explicit-sharding-support branch 156 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone