Skip to content

Commit 6426c9d

Browse files
committed
Merge branch 'main' of https://github.com/Kaggle/docker-python into lsp-versions
2 parents d39726c + 170210f commit 6426c9d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tpu/config.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@ PYTHON_VERSION_PATH=python3.10
55
TENSORFLOW_VERSION=2.15.0
66
TF_LIBTPU_VERSION=1.9.0
77
TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64
8-
JAX_VERSION=0.4.23
8+
JAX_VERSION=0.4.26
9+
# gsutil ls gs://pytorch-xla-releases/wheels/tpuvm/* | grep libtpu | grep -v -E ".*rc[0-9].*"
910
# Supports nightly
1011
TORCH_VERSION=2.2.0
1112
# https://github.com/pytorch/audio supports nightly

0 commit comments

Comments
 (0)