flax
0453d38f - Merge pull request #2730 from IvyZX:pjit

Commit

3 years ago

Merge pull request #2730 from IvyZX:pjit PiperOrigin-RevId: 497260943

References

#2730 - Guide for Flax partitioning (auto SPMD) API

Author

a-googler

a-googler

Parents

Loading