flax
47626d88 - Added sharding propagation support in nnx.eval_shape

Commit

69 days ago

Added sharding propagation support in nnx.eval_shape

References

eval-shape-sharding

#5111 - Added sharding propagation support in nnx.eval_shape

Author

vfdev-5

vfdev-5

Committer

vfdev-5

vfdev-5

Parents

Loading