Johnny YangandChengji Yao
8bed179109
Signed-off-by: Johnny Yang <johnnyyang@google.com> Co-authored-by: Chengji Yao <chengjiyao@google.com>
17 lines
248 B
Plaintext
17 lines
248 B
Plaintext
# 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_info==0.4.0
|
|
|
|
# Install torch_xla
|
|
torch_xla[tpu, pallas]==2.8.0 |