mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00

I forgot that the default setting is actually in jaxlib:
fbe9a80fdb/xla/python/xla_client.py (L135)
To be able to make this change as a jax-only release, I manually set
the env var on Cloud TPU if it isn't already set.