5 ms·there is limited TPU support in pytorch via torch_xlaby hustwindmaple1 2y agothere is limited TPU support in pytorch via torch_xlajdeaton 2y agoSounds limited