From 59f75d538b2ad1579a56c52d6d2ff7f9585b93fc Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 7 Jun 2021 13:04:18 +0100 Subject: [PATCH] [JAX] Bump jax lib (#12053) * fix_torch_device_generate_test * remove @ * bump up jax lib --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 9e1949f33f..4fe1672e0a 100644 --- a/setup.py +++ b/setup.py @@ -104,7 +104,7 @@ _deps = [ "ipadic>=1.0.0,<2.0", "isort>=5.5.4", "jax>=0.2.8", - "jaxlib>=0.1.59", + "jaxlib>=0.1.65", "jieba", "keras2onnx", "nltk",