2024.11.1 Release

(*) = 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

2024.10.1 Release

• 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

Version 2024.10.1b1 pre-release

v2024.10.1b1

2024.10.1b1 pre-release