jax
aeac6b03 - Fix pmap with sharded typed prng key

Commit

1 year ago

Fix pmap with sharded typed prng key PiperOrigin-RevId: 714293671

References

#25845 - Fix pmap with sharded typed prng key

#213 - CI: 01/27/25 upstream sync

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading