flax
Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps
#3775
Merged

Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps #3775

ghost merged 1 commit into main from test_617511370
copybara-service
copybara-service copybara-service force pushed 1 year ago
superbobry Access thread_resources via jax.interpreters.pxla instead of jax.expe…
e1465fed
copybara-service copybara-service force pushed to e1465fed 1 year ago
copybara-service copybara-service closed this 1 year ago
copybara-service copybara-service deleted the test_617511370 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone