rocm_jax/docs/requirements.txt

25 lines
588 B
Plaintext

absl-py
ipython>=8.8.0 # 8.7.0 has ipython3 lexer error
pydata-sphinx-theme==0.14.4 # v0.15 breaks sidebar toggling
sphinx>=7.3.2,<8.0 # 7.3.0 breaks sphinx-book-theme; 8.0 breaks myst-nb 1.1
sphinx-book-theme==1.1.1 # v1.1.2 requires pydata-sphinx-theme v0.15
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
sphinx-design
sphinxext-rediraffe
myst-nb>=1.0.0
# Packages used for CI tests.
flatbuffers
pytest
pytest-xdist
# Packages used for notebook execution
matplotlib
scikit-learn
pooch
numpy
rich[jupyter]
cmake
.[ci] # Install jax from the current directory; jaxlib from pypi.