Google open-sources TorchTPU, a native PyTorch back end for TPUs

Posted 5 hours ago by ilreb
204 points

https://github.com/google-pytorch/torch_tpu

13 comments

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..