2024.10.1 Release

jakevdp released this 01 Oct 18:55

· 85 commits to main since this release

9c03df0

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

Assets 2