Releases: jax-ml/jax-ai-stack
Releases · jax-ml/jax-ai-stack
2025.1.9 Release
jax-ai-stackpackages:jax==0.4.38↗️ flax==0.10.2ml_dtypes==0.4.0optax==0.2.4orbax-checkpoint==0.11.0↗️ orbax-export==0.0.6
jax-ai-stack[tfds]packages:tensorflow==2.18.0tensorflow_datasets==4.9.7
jax-ai-stack[grain]packages:grain==0.2.3↗️
2024.12.10 Release
jax-ai-stackpackages:jax==0.4.37↗️ flax==0.10.2↗️ ml_dtypes==0.4.0optax==0.2.4↗️ orbax-checkpoint==0.10.2↗️ orbax-export==0.0.6↗️
jax-ai-stack[tfds]packages:tensorflow==2.18.0tensorflow_datasets==4.9.7↗️
jax-ai-stack[grain]packages:grain==0.2.2
2024.11.1 Release
jax-ai-stackpackages:jax==0.4.35↗️ flax==0.10.0↗️ ml_dtypes==0.4.0optax==0.2.3orbax-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↗️
2024.10.1 Release
jax-ai-stackpackages:jax==0.4.33flax==0.9.0ml_dtypes==0.4.0optax==0.2.3orbax-checkpoint==0.6.4orbax-export==0.0.5
jax-ai-stack[tfds]packages:tensorflow==2.17.0tensorflow_datasets==4.9.6
jax-ai-stack[grain]packages:grain==0.2.1
Version 2024.10.1b1 pre-release
v2024.10.1b1 2024.10.1b1 pre-release