* fix_torch_device_generate_test * remove @ * bump up jax lib
Attach a file by drag & drop or click to upload
