numpy

[dev]
pytest
pytest-cov
pre-commit
sphinx==6.1.3
sphinx_rtd_theme==1.2.0
sphinx-autoapi==2.0.1
astroid<3
black
nbconvert
nbsphinx
ipython
ipykernel
matplotlib
scikit-learn
numpy

[jax]
jax

[torch]
torch
