flax
0399b9d3 - Generalize out_sharding to work with NamedSharding and Format

Commit

15 days ago

Generalize out_sharding to work with NamedSharding and Format

References

#5246 - Generalize out_sharding to work with NamedSharding and Format

Author

samanklesaria

samanklesaria

Committer

samanklesaria

samanklesaria

Parents

Loading