9 ms·
We have been working with G on TPU support for PyTorch. Have you tried it out? https://github.com/pytorch/xla https://github.com/pytorch/xla
by jspisak 7y ago
We have been working with G on TPU support for PyTorch. Have you tried it out? https://github.com/pytorch/xla https://github.com/pytorch/xla