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

[dev]
mkdocs>=1.6.0
mkdocs-material>=9.7.0
mkdocstrings-python>=1.10.0
click>=8.2.0
tensorboard>=2.20.0
tensorboard_plugin_profile>=2.20.0
xprof>=2.20.0

[gpu]
jax[cuda12]>=0.4.0
