rocm_jax/docs/requirements.txt
2022-08-08 16:11:19 +00:00

18 lines
327 B
Plaintext

sphinx >=3
sphinx-autodoc-typehints
sphinx-book-theme>=0.3.3
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
jupyter-sphinx>=0.3.2
myst-nb
# Packages used for CI tests.
pytest
pytest-xdist
# Packages used for notebook execution
matplotlib
scikit-learn
numpy
.[ci] # Install jax from the current directory; jaxlib from pypi.