TorchTPU: Running PyTorch Natively on TPUs at Google Scale

Posted 16 hours ago by mji
152 points

https://developers.googleblog.com/torchtpu-running-pytorch-natively-on-tpus-at-google-scale/

11 comments

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..

Loading..