Google, PyTorch를 TPU에서 그대로 돌리는 TorchTPU 공개

Google open-sources TorchTPU, a native PyTorch back end for TPUs

Google이 TPU용 네이티브 PyTorch 백엔드인 TorchTPU를 오픈소스로 공개했다. TPU에서 PyTorch 모델과 연산을 실행할 수 있게 해주는 ATen 커널 구현과 XLA 연산 최적화용 컴파일 캐시를 포함하며, 데이터·텐서 병렬 분산 학습과 torch.compile() 그래프 모드 실행을 지원한다. Llama, Qwen, ResNet, DLRM, minGPT, UNet 예제와 Marimo 노트북, 벤치마킹 도구도 함께 제공된다. pip로 0.2.0 버전을 설치할 수 있고 Apache 2.0 라이선스로 배포된다.