# sphinx <4 required by myst-nb v0.12.0 (Feb 2021)
# sphinx >=3 required by sphinx-autodoc-typehints v1.11.1 (Oct 2020)
sphinx >=3, <4
sphinx_rtd_theme
sphinx-autodoc-typehints==1.11.1
jupyter-sphinx>=0.3.2
myst-nb

# Packages used for CI tests.
pytest
pytest-xdist

# Packages used for notebook execution
matplotlib
scikit-learn
# TODO(jakevdp) remove numpy pinning when minimum jaxlib supports newer numpy.
numpy<1.21
.[cpu]  # Install jax from the current directory; jaxlib from pypi.