2024.12.10 Release

jakevdp released this 10 Dec 22:05

· 8 commits to main since this release

a2204fd

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

Assets 2