rocm_jax/docs/requirements.txt
2020-05-11 14:37:17 +03:00

15 lines
223 B
Plaintext

sphinx
sphinx_rtd_theme
jaxlib
ipykernel
nbsphinx
sphinx-autodoc-typehints
# For host_callback.py
msgpack
# The next packages are for notebooks
matplotlib
sklearn
# Must install jax itself for notebook execution to work
.