3 ms·
Last I've checked (a year or two ago) PyTorch support for TPU's were atrocious. Have they gotten any better?
by Learnedvector 5y ago
Last I've checked (a year or two ago) PyTorch support for TPU's were atrocious. Have they gotten any better?
- ypcx 5y agohttps://github.com/pytorch/xla/ https://github.com/pytorch/xla/
- tubby12345 5y agoPyTorch XLA is mature backend. In fact several other accelerators support PyTorch by lowering from XLA.