jax==0.8.1
jaxlib==0.8.1
diffrax>=0.7.0
equinox>=0.13.2
numpy>=2.0.0
chex>=0.1.91
optax>=0.2.3
tqdm>=4.65
h5py>=3.10

[dev]
pytest>=8.0
pytest-xdist>=3.5
ruff>=0.4
ty>=0.0.20
pre-commit>=3.7

[docs]
mkdocs>=1.6
mkdocs-material>=9.5
mkdocstrings[python]>=0.25
pymdown-extensions>=10.0
