flax
70351571 - Added sharding propagation support in nnx.eval_shape

Commit

2 days ago

Added sharding propagation support in nnx.eval_shape

References

#5247 - Add sharding propagation support in nnx.eval_shape (clone of #5111)

Author

vfdev-5

vfdev-5

Committer

samanklesaria

samanklesaria

Parents

Loading