Merge pull request jax-ml#9373 from skye:update_libtpu_version

PiperOrigin-RevId: 424994481
spundun · Jan 29, 2022 · abc339b · abc339b
2 parents f3ae2c0 + d096d9a
commit abc339b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -20,7 +20,7 @@
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '805']
 _default_cudnn_version = '82'
-_libtpu_version = '0.1.dev20211208'
+_libtpu_version = '0.1.dev20220128'
 
 _dct = {}
 with open('jax/version.py') as f: