mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
22 lines
230 B
Plaintext
22 lines
230 B
Plaintext
#
|
|
# test deps
|
|
#
|
|
-r test-requirements.txt
|
|
|
|
#
|
|
# build deps
|
|
#
|
|
numpy~=2.0.0; python_version<="3.12"
|
|
numpy~=2.1.0; python_version>="3.13"
|
|
|
|
#
|
|
# runtime deps
|
|
#
|
|
scipy>=1.13.1
|
|
|
|
ml_dtypes>=0.4.0
|
|
opt_einsum
|
|
zstandard
|
|
etils[epath]
|
|
setuptools
|