rocm_jax/build/requirements.in

#
# 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
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
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`
			`#`
Add Python 3.130rc2 support to the build. This PR depends on https://github.com/openxla/xla/pull/17169. The change does not fail existing builds, but to be able to use python 3.13 functionality in jax the corresponding XLA pr needs to land first and get integrated with JAX (happens automatically). PiperOrigin-RevId: 675243989 2024-09-16 12:13:47 -07:00			`numpy~=2.0.0; python_version<="3.12"`
			`numpy~=2.1.0; python_version>="3.13"`
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`
			`#`
Add Python 3.130rc2 support to the build. This PR depends on https://github.com/openxla/xla/pull/17169. The change does not fail existing builds, but to be able to use python 3.13 functionality in jax the corresponding XLA pr needs to land first and get integrated with JAX (happens automatically). PiperOrigin-RevId: 675243989 2024-09-16 12:13:47 -07: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]`
Add CI jobs for python 3.13.0rc2. PiperOrigin-RevId: 675758096 2024-09-17 16:50:55 -07:00			`# TODO(ybaturina): remove setuptools version`
			`setuptools<71.0.0`