jax>=0.4.0
numpy>=1.20.0
scikit-learn>=1.0.0
optax>=0.1.0
pandas>=1.3.0

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

[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
seaborn>=0.13.2

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