rocm_jax/build/requirements.in

#
# 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

#
# runtime deps
#
scipy~=1.13.1

ml_dtypes>=0.4.0
opt_einsum
zstandard
etils[epath]
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details PiperOrigin-RevId: 634146391 2024-05-15 18:20:14 -07:00			`#`
			`# test deps`
			`#`
			`-r test-requirements.txt`
Pin matplotlib < 3.9 for Python 3.10 and earlier. matplotlib 3.9.0 pins NumPy 1.23 or newer, which is incompatible with our minimum Numpy pin. 2024-05-22 15:07:09 +00:00
			`# 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"`
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details PiperOrigin-RevId: 634146391 2024-05-15 18:20:14 -07:00
			`#`
			`# build deps`
			`#`
Use NumPy 2.0.0 and SciPy 1.13.1 in builds. Don't override the XLA repository in the nightly Windows CI builds, which should be building JAX as it exists in the source repository. 2024-06-17 19:17:12 +00:00			`numpy~=2.0.0`
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details PiperOrigin-RevId: 634146391 2024-05-15 18:20:14 -07:00
			`#`
			`# runtime deps`
			`#`
Use NumPy 2.0.0 and SciPy 1.13.1 in builds. Don't override the XLA repository in the nightly Windows CI builds, which should be building JAX as it exists in the source repository. 2024-06-17 19:17:12 +00:00			`scipy~=1.13.1`
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details PiperOrigin-RevId: 634146391 2024-05-15 18:20:14 -07:00
			`ml_dtypes>=0.4.0`
			`opt_einsum`
			`zstandard`
			`etils[epath]`