Skip to content

Tags: jax-ml/jax

Tags

jax-v0.11.2

Toggle jax-v0.11.2's commit message
Prepare for JAX release 0.11.2

jax-v0.11.1

Toggle jax-v0.11.1's commit message
Prepare for JAX release 0.11.1

jax-v0.11.0

Toggle jax-v0.11.0's commit message
Prepare for JAX release 0.11.0

jax-v0.10.2

Toggle jax-v0.10.2's commit message
Disable gpu_pallas_distributed_nvshmem_test_gpu which was failing at the

branch cut point.

jax-v0.10.1

Toggle jax-v0.10.1's commit message

Unverified

This user has not yet uploaded their public signing key.
Apply patch to revert XLA PR #42396 on release/0.10.1 branch

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