mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
108 lines
1.8 KiB
Plaintext
108 lines
1.8 KiB
Plaintext
--pre
|
|
--extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
|
|
numpy
|
|
|
|
--pre
|
|
--extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
|
|
scipy
|
|
|
|
absl-py==2.1.0
|
|
|
|
attrs==24.3.0
|
|
|
|
auditwheel==6.2.0
|
|
|
|
build==1.2.2.post1
|
|
|
|
cloudpickle==3.1.1 # version 3.1.0 leads to recursion error
|
|
|
|
colorama==0.4.6
|
|
|
|
contourpy==1.3.1
|
|
|
|
cycler==0.12.1
|
|
|
|
etils[epath,epy]==1.11.0
|
|
|
|
execnet==2.1.1
|
|
|
|
filelock==3.16.1
|
|
|
|
flatbuffers==24.12.23
|
|
|
|
fonttools==4.56.0
|
|
|
|
fsspec==2024.12.0
|
|
|
|
hypothesis==6.123.9
|
|
|
|
importlib-resources==6.5.2
|
|
|
|
iniconfig==2.0.0
|
|
|
|
kiwisolver==1.4.8
|
|
|
|
markdown-it-py==3.0.0
|
|
|
|
matplotlib==3.10.1
|
|
|
|
mdurl==0.1.2
|
|
|
|
ml-dtypes==0.5.1
|
|
|
|
mpmath==1.3.0
|
|
|
|
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux"
|
|
|
|
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux"
|
|
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux"
|
|
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux"
|
|
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux"
|
|
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux"
|
|
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux"
|
|
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux"
|
|
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux"
|
|
|
|
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux"
|
|
opt-einsum==3.4.0
|
|
|
|
packaging==24.2
|
|
|
|
pillow==11.1.0
|
|
pluggy==1.5.0
|
|
|
|
portpicker==1.6.0
|
|
|
|
psutil==6.1.1
|
|
pyelftools==0.31
|
|
|
|
pygments==2.19.1
|
|
|
|
pyparsing==3.2.2 # version 3.2.1 fails with SyntaxError(originally SyntaxWarning): 'return' in a 'finally' block in pyparsing/core.py", line 5716
|
|
|
|
pyproject-hooks==1.2.0
|
|
|
|
pytest==8.3.4
|
|
|
|
pytest-xdist==3.6.1
|
|
|
|
python-dateutil==2.9.0.post0
|
|
|
|
rich==13.9.4
|
|
|
|
six==1.17.0
|
|
|
|
sortedcontainers==2.4.0
|
|
|
|
typing-extensions==4.12.2
|
|
|
|
wheel==0.45.1
|
|
|
|
zipp==3.21.0
|
|
|
|
# python 3.14t can't compile 0.23.0
|
|
# due to https://github.com/indygreg/python-zstandard/issues/231
|
|
# zstandard==0.23.0
|
|
|
|
setuptools==70.3.0
|