We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents d39726c + 170210f commit 6426c9dCopy full SHA for 6426c9d
tpu/config.txt
@@ -5,7 +5,8 @@ PYTHON_VERSION_PATH=python3.10
5
TENSORFLOW_VERSION=2.15.0
6
TF_LIBTPU_VERSION=1.9.0
7
TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64
8
-JAX_VERSION=0.4.23
+JAX_VERSION=0.4.26
9
+# gsutil ls gs://pytorch-xla-releases/wheels/tpuvm/* | grep libtpu | grep -v -E ".*rc[0-9].*"
10
# Supports nightly
11
TORCH_VERSION=2.2.0
12
# https://github.com/pytorch/audio supports nightly
0 commit comments