3 ms·
PyTorch already has TPU support via XLA and has been used in research at the scale of billions of parameters. https://github.com/pytorch/xla https://github.com
by Smerity 7y ago
PyTorch already has TPU support via XLA and has been used in research at the scale of billions of parameters.
https://github.com/pytorch/xla https://github.com/pytorch/xla