flax
Implement jax_utils.replicate using jax.device_put_replicated
#1579

Merged

Implement jax_utils.replicate using jax.device_put_replicated #1579

copybara-service merged 1 commit into main from test_399375769

copybara-service force pushed from 3ca28487 4 years ago

google-cla added cla: yes

Implement jax_utils.replicate using jax.device_put_replicated

97dde9ab

copybara-service force pushed to 97dde9ab 4 years ago

copybara-service merged 97dde9ab into main 4 years ago

copybara-service deleted the test_399375769 branch 4 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone