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

google-research / t5x

Apache License 2.0

2.58k stars 296 forks source link

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

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

The maps submodule is deprecated and will be removed soon.