jax<1.0,>=0.5.2
flax<1.0,>=0.12.0
optax<1.0,>=0.2.7
jaxtyping<1.0,>=0.3.0
numpy<3.0,>=1.26.4
matplotlib<4.0,>=3.10.7
h5py<4.0,>=3.15.0
tqdm<5.0,>=4.66.4

[cuda12]
jax[cuda12]<1.0,>=0.5.2

[dev]
build<3.0,>=1.0
twine<6,>=5
ruff<1.0,>=0.6
pytest<9,>=8
