rocm_jax/docs/requirements.txt
2021-02-16 13:25:59 -08:00

20 lines
424 B
Plaintext

# sphinx <3 is required by myst-parser[sphinx] v0.9.1, the latest version as of 2020-08-05
sphinx >=2.1, <3
sphinx_rtd_theme
# sphinx-autodoc-typehints 1.11 requires sphinx >= 3, but we can't have that right now.
sphinx-autodoc-typehints <1.11
jaxlib
ipykernel
myst-nb
# The next packages are for notebooks
matplotlib
sklearn
# For CI tests.
pytest
pytest-xdist
# Must install jax itself for notebook execution to work
.