jax>=0.4.0
matplotlib>=3.10.3
numpy>=1.20.0
optax>=0.1.0

[dev]
ruff>=0.9.0
pytest>=7.0.0
black==23.7.0
mypy>=1.8.0
types-requests
types-setuptools

[docs]
sphinx>=7.2.0
sphinx-rtd-theme>=2.0.0
sphinx-autodoc-typehints>=1.25.0
myst-parser>=2.0.0
nbsphinx>=0.9.0
matplotlib>=3.7.5

[examples]
matplotlib>=3.7.5
pandas>=1.3.0
seaborn>=0.13.2
jupyter>=1.0.0
