flax
dc62cb8e - Migrate jax.host_id to jax.process_index and jax.host_count to jax.process_count.

Commit

4 years ago

Migrate jax.host_id to jax.process_index and jax.host_count to jax.process_count. PiperOrigin-RevId: 379444367

References

Author

tutmann

Committer

a-googler

Parents