Hacker News new | ask | show | jobs
by Learnedvector 1710 days ago
Last I've checked (a year or two ago) PyTorch support for TPU's were atrocious. Have they gotten any better?
2 comments
tubby12345 1710 days ago
PyTorch XLA is mature backend. In fact several other accelerators support PyTorch by lowering from XLA.
link
ypcx 1710 days ago
https://github.com/pytorch/xla/
link