flax
Use jax.api.device_put_sharded() in place of private JAX APIs.
#466

Merged

Use jax.api.device_put_sharded() in place of private JAX APIs. #466

copybara-service merged 1 commit into master from test_332079924

google-cla added cla: yes

copybara-service force pushed 5 years ago

Use jax.api.device_put_sharded() in place of private JAX APIs.

c2ba4a38

copybara-service force pushed to c2ba4a38 5 years ago

copybara-service merged c2ba4a38 into master 5 years ago

copybara-service deleted the test_332079924 branch 5 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone