flax
332b7d04 - Merge pull request #3013 from IvyZX:pjit

Commit

2 years ago

Merge pull request #3013 from IvyZX:pjit PiperOrigin-RevId: 522473275

References

#3013 - Fix `get_partition_spec` on replicated array

Author

a-googler

a-googler

Parents

Loading