mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
28 lines
448 B
Plaintext
28 lines
448 B
Plaintext
#
|
|
# test deps
|
|
#
|
|
-r test-requirements.txt
|
|
|
|
# matplotlib 3.9.0 pins NumPy 1.23, which is incompatible with the requirement
|
|
# below.
|
|
matplotlib~=3.8.4; python_version<="3.10"
|
|
matplotlib; python_version>="3.11"
|
|
|
|
#
|
|
# 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
|