Google open-sourced TorchTPU, so PyTorch on TPUs just got a lot easier

T2Covered by 1 source · first reported by Help Net Security at 21 Oct, 03:42 UTC

Google released TorchTPU as open source, giving PyTorch developers a native way to run models on the company’s Tensor Processing Units. Google also announced TPU support in the vLLM and SGLang serving engines, the TorchTitan training framework, Ray, and Hugging Face’s libraries. Teams that write…