Skip to content

Tags: jax-ml/jax

Tags

jax-v0.10.0

Toggle jax-v0.10.0's commit message 
Add a libtpu version guard in test_dce_sink_prevents_xla_dce

PiperOrigin-RevId: 900420405

jax-v0.9.2

Toggle jax-v0.9.2's commit message 
Prepare for JAX release 0.9.2

jax-v0.9.1

Toggle jax-v0.9.1's commit message 
Prepare for JAX release 0.9.1

jax-v0.9.0.1

Toggle jax-v0.9.0.1's commit message 
Prepare for JAX release 0.9.0.1

jax-v0.8.3

Toggle jax-v0.8.3's commit message 
Prepare for JAX release 0.8.3

jax-v0.9.0

Toggle jax-v0.9.0's commit message 
Use maxsize=None with trace_to_jaxpr's weakref_lru_cache to get more …

…cache hits. This also restores previous behavior of lu.cache.

PiperOrigin-RevId: 858211335

jax-v0.8.2

Toggle jax-v0.8.2's commit message 
Prepare for JAX release 0.8.2

jax-v0.8.1

Toggle jax-v0.8.1's commit message 
Prepare for JAX release 0.8.1

jax-v0.8.0

Toggle jax-v0.8.0's commit message 
Reverts f44bc6e

PiperOrigin-RevId: 819903461

jax-v0.7.2

Toggle jax-v0.7.2's commit message 
Prepare for JAX release 0.7.2