# CPU workshop reference. Install GPU/TPU JAX separately using official target instructions.
jax==0.9.2
flax==0.12.6
optax==0.2.8
numpy==2.4.4
matplotlib==3.10.8
