jax
jaxlib
numpy>=1.22
scipy>=1.9
scikit-learn
pandas
jax-tqdm
tqdm

[doc]
jax
jaxlib
numpy>=1.22
scipy>=1.9
scikit-learn
pandas
jax-tqdm
tqdm
sphinx!=4.1.0
sphinx-gallery
sphinx_design
sphinx-book-theme
sphinxcontrib-bibtex
sphinx_contributors
numpydoc
matplotlib
xgi

[full]
jax
jaxlib
numpy>=1.22
scipy>=1.9
scikit-learn
pandas
jax-tqdm
tqdm
pytest
pytest-cov
codecov
xarray
sphinx!=4.1.0
sphinx-gallery
sphinx_design
sphinx-book-theme
sphinxcontrib-bibtex
sphinx_contributors
numpydoc
matplotlib
xgi
flake8
pep8-naming
black

[lint]
jax
jaxlib
numpy>=1.22
scipy>=1.9
scikit-learn
pandas
jax-tqdm
tqdm
flake8
pep8-naming
black

[test]
jax
jaxlib
numpy>=1.22
scipy>=1.9
scikit-learn
pandas
jax-tqdm
tqdm
pytest
pytest-cov
codecov
xarray
