flax
e8a14af9 - Merge pull request #3580 from google:nnx-get-named-sharding

Commit

1 year ago

Merge pull request #3580 from google:nnx-get-named-sharding PiperOrigin-RevId: 595460697

References

Author

a-googler

Parents