Change log

↗️ = updated since previous release

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