diff --git a/setup.py b/setup.py index 48ec4df26a1a..9efd4f734a02 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,7 @@ _default_cuda_version = '11' _available_cudnn_versions = ['82', '805'] _default_cudnn_version = '82' -_libtpu_version = '0.1.dev20220329' +_libtpu_version = '0.1.dev20220407' _dct = {} with open('jax/version.py') as f: