jax>=0.4.0
diffrax>=0.7.0
equinox>=0.13.0
h5py>=3.14.0
tqdm>=4.67.0
matplotlib>=3.10.0

[cuda12]
jax[cuda12]>=0.4.0

[dev]
mkdocs>=1.6.0
mkdocs-material>=9.7.0
mkdocstrings-python>=1.10.0
pytest>=8.0

[docs]
mkdocs>=1.6.0
mkdocs-material>=9.7.0
mkdocstrings-python>=1.10.0

[gpu]
jax[cuda13]>=0.4.0

[profiling]
click>=8.2.0
tensorboard>=2.20.0
tensorboard_plugin_profile>=2.20.0
xprof>=2.20.0

[test]
pytest>=8.0
