flax
e1465fed - Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps

Commit

1 year ago

Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps The maps submodule is deprecated and will be removed soon. PiperOrigin-RevId: 617572444

References

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

Author

superbobry

Committer

a-googler

Parents

006da41a

flax e1465fed - Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps

flax
e1465fed - Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps