mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
23 lines
282 B
Plaintext
23 lines
282 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]
|
|
# TODO(ybaturina): remove setuptools version
|
|
setuptools<71.0.0
|