(*) = updated since previous release
- `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`