flax
Added sharding propagation support in nnx.eval_shape
#5111

Open

Added sharding propagation support in nnx.eval_shape #5111

vfdev-5 wants to merge 1 commit into main from eval-shape-sharding

Added sharding propagation support in nnx.eval_shape

47626d88

vfdev-5 force pushed from f717ff6d to 47626d88 49 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone