rocm_jax/docs/requirements.txt

18 lines
404 B
Plaintext

# 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
numpy
.[cpu] # Install jax from the current directory; jaxlib from pypi.