flax
Migrate jax.host_id to jax.process_index and jax.host_count to jax.process_count.
#1363

Merged

Migrate jax.host_id to jax.process_index and jax.host_count to jax.process_count. #1363

copybara-service merged 1 commit into master from test_379310749

google-cla added cla: yes

copybara-service force pushed 5 years ago

copybara-service force pushed 5 years ago

Migrate jax.host_id to jax.process_index and jax.host_count to jax.pr…

dc62cb8e

copybara-service force pushed to dc62cb8e 5 years ago

copybara-service merged dc62cb8e into master 5 years ago

copybara-service deleted the test_379310749 branch 5 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone