mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
23 lines
526 B
Plaintext
23 lines
526 B
Plaintext
absl-py
|
|
ipython>=8.8.0 # 8.7.0 has ipython3 lexer error
|
|
sphinx>=7.3.2,<8.0 # 7.3.0 breaks sphinx-book-theme; 8.0 breaks myst-nb 1.1
|
|
sphinx-book-theme>=1.0.1 # Older versions fail to pin pydata-sphinx-theme
|
|
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
|
|
numpy
|
|
rich[jupyter]
|
|
cmake
|
|
.[ci] # Install jax from the current directory; jaxlib from pypi.
|