122 lines (112 loc) · 3.09 KB

Change log

↗️ = updated since previous release

v2025.10.28

jax-ai-stack packages:
- jax==0.8.0 ↗️
- chex==0.1.91 ↗️
- grain==0.2.13 ↗️
- flax==0.12.0 ↗️
- ml_dtypes==0.5.3
- optax==0.2.6 ↗️
- orbax-checkpoint==0.11.26 ↗️
- orbax-export==0.0.8 ↗️
jax-ai-stack[tfds] packages:
- tensorflow==2.20.0
- tensorflow_datasets==4.9.9

v2025.9.3

jax-ai-stack packages:
- jax==0.7.0 ↗️
- chex==0.1.90 ↗️
- grain==0.2.12 ↗️
- flax==0.11.1 ↗️
- ml_dtypes==0.5.3 ↗️
- optax==0.2.5 ↗️
- orbax-checkpoint==0.11.23 ↗️
- orbax-export==0.0.6
jax-ai-stack[tfds] packages:
- tensorflow==2.20.0 ↗️
- tensorflow_datasets==4.9.9 ↗️
jax-ai-stack[grain] has been removed and is now part of the standard package

v2025.4.9

jax-ai-stack packages:
- jax==0.5.3 ↗️
- chex==0.1.89 ↗️
- flax==0.10.5 ↗️
- ml_dtypes==0.5.1 ↗️
- optax==0.2.4
- orbax-checkpoint==0.11.12 ↗️
- orbax-export==0.0.6
jax-ai-stack[tfds] packages:
- tensorflow==2.19.0 ↗️
- tensorflow_datasets==4.9.8 ↗️
jax-ai-stack[grain] packages:
- grain==0.2.7 ↗️

v2025.2.5

jax-ai-stack packages:
- jax==0.5.0 ↗️
- chex==0.1.88 🆕
- flax==0.10.2
- ml_dtypes==0.4.0
- optax==0.2.4
- orbax-checkpoint==0.11.1 ↗️
- orbax-export==0.0.6
jax-ai-stack[tfds] packages:
- tensorflow==2.18.0
- tensorflow_datasets==4.9.7
jax-ai-stack[grain] packages:
- grain==0.2.3

Notes

This version of jax-ai-stack adds chex as a direct pinned dependency. Prior to this release chex was an indirect unpinned dependency via optax.

v2025.1.9

jax-ai-stack packages:
- jax==0.4.38 ↗️
- flax==0.10.2
- ml_dtypes==0.4.0
- optax==0.2.4
- orbax-checkpoint==0.11.0 ↗️
- orbax-export==0.0.6
jax-ai-stack[tfds] packages:
- tensorflow==2.18.0
- tensorflow_datasets==4.9.7
jax-ai-stack[grain] packages:
- grain==0.2.3 ↗️

v2024.12.10

jax-ai-stack packages:
- jax==0.4.37 ↗️
- flax==0.10.2 ↗️
- ml_dtypes==0.4.0
- optax==0.2.4 ↗️
- orbax-checkpoint==0.10.2 ↗️
- orbax-export==0.0.6 ↗️
jax-ai-stack[tfds] packages:
- tensorflow==2.18.0
- tensorflow_datasets==4.9.7 ↗️
jax-ai-stack[grain] packages:
- grain==0.2.2

v2024.11.1

jax-ai-stack packages:
- jax==0.4.35 ↗️
- flax==0.10.0 ↗️
- ml_dtypes==0.4.0
- optax==0.2.3
- orbax-checkpoint==0.7.0 ↗️
- orbax-export==0.0.5
jax-ai-stack[tfds] packages:
- tensorflow==2.18.0 ↗️
- tensorflow_datasets==4.9.6
jax-ai-stack[grain] packages:
- grain==0.2.2 ↗️

v2024.10.1

jax-ai-stack packages:
- jax==0.4.33
- flax==0.9.0
- ml_dtypes==0.4.0
- optax==0.2.3
- orbax-checkpoint==0.6.4
- orbax-export==0.0.5
jax-ai-stack[tfds] packages:
- tensorflow==2.17.0
- tensorflow_datasets==4.9.6
jax-ai-stack[grain] packages:
- grain==0.2.1