[JAX] Bump jax lib (#12053)
* fix_torch_device_generate_test * remove @ * bump up jax lib
This commit is contained in:
parent
185122ef22
commit
59f75d538b
* fix_torch_device_generate_test * remove @ * bump up jax lib