numpy
scipy
pandas
polars
xarray
netcdf4
pyyaml
setuptools
jax>=0.4.38
diffrax>=0.6.2
equinox>=0.11.10
optimistix>=0.0.7

[docs]
mkdocs>=1.6
mkdocs-material>=9.5
mkdocstrings[python]>=0.25
pymdown-extensions>=10.7
ruff>=0.4
matplotlib>=3.5
jax[cpu]>=0.4.38
diffrax>=0.6.2
equinox>=0.11.10
optimistix>=0.0.7

[examples]
matplotlib>=3.5

[gpu]
jax[cuda12]>=0.4.38
diffrax>=0.6.2
equinox>=0.11.10
optimistix>=0.0.7

[test]
pytest>=7
pytest-xdist>=3
hypothesis>=6
pyrefly>=1.1
