requirements/tpu.txt

# Common dependencies
-r common.txt

# Dependencies for TPU
cmake>=3.26.1
packaging>=24.2
setuptools-scm>=8
wheel
jinja2>=3.1.6
ray[default]
ray[data]
setuptools==78.1.0
nixl==0.3.0
tpu-inference==0.12.0
[Hardware] Initial TPU integration (#5292) 2024-06-12 11:53:03 -07:00			`# Common dependencies`
Move requirements into their own directory (#12547) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> 2025-03-08 17:44:35 +01:00			`-r common.txt`
[Hardware] Initial TPU integration (#5292) 2024-06-12 11:53:03 -07:00
			`# Dependencies for TPU`
[Bugfix] Use cmake 3.26.1 instead of 3.26 to avoid build failure (#19019) Signed-off-by: Lu Fang <lufang@fb.com> 2025-06-03 15:16:17 +08:00			`cmake>=3.26.1`
[Build] Require setuptools >= 77.0.3 for PEP 639 (#17389) Signed-off-by: Russell Bryant <rbryant@redhat.com> 2025-05-01 02:25:36 -04:00			`packaging>=24.2`
Refactor TPU requirements file and pin build dependencies (#10010) Signed-off-by: Richard Liu <ricliu@google.com> 2024-11-05 08:48:44 -08:00			`setuptools-scm>=8`
			`wheel`
[Build/CI] Upgrade jinja2 to get 3 moderate CVE fixes (#14839) Signed-off-by: Russell Bryant <rbryant@redhat.com> 2025-03-15 01:38:19 -04:00			`jinja2>=3.1.6`
[TPU] Support single and multi-host TPUs on GKE (#7613) 2024-08-30 00:27:40 -07:00			`ray[default]`
Add ray[data] as tpu dependency (#14691) Signed-off-by: <ricliu@google.com> Signed-off-by: Richard Liu <ricliu@google.com> 2025-03-12 20:13:48 -07:00			`ray[data]`
[CI] Uses Python 3.11 for TPU (#17359) Signed-off-by: Aaron Pham <contact@aarnphm.xyz> 2025-04-29 13:39:16 -04:00			`setuptools==78.1.0`
[P/D] Support CPU Transfer in NixlConnector (#18293) Signed-off-by: Juncheng Gu <juncgu@gmail.com> Signed-off-by: Richard Liu <ricliu@google.com> Co-authored-by: Richard Liu <39319471+richardsliu@users.noreply.github.com> Co-authored-by: Richard Liu <ricliu@google.com> 2025-07-24 09:58:42 -07:00			`nixl==0.3.0`
[TPU] Bump tpu-inference to 0.12.0 (#30221) Signed-off-by: Johnny Yang <johnnyyang@google.com> 2025-12-08 12:10:10 -08:00			`tpu-inference==0.12.0`