mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 04:46:06 +00:00
21 lines
420 B
Plaintext
21 lines
420 B
Plaintext
absl-py
|
|
# TODO(sharadmv,jakevdp): fix IPython3 issue and remove pinned version
|
|
ipython<=8.6.0
|
|
sphinx>=4
|
|
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.
|