Google, TPU용 네이티브 PyTorch 백엔드 TorchTPU 오픈소스로 공개

TorchTPU는 Google의 TPU에서 PyTorch 모델과 연산을 실행할 수 있게 하는 백엔드임
TPU용 ATen 커널과 XLA 연산 실행을 최적화하는 컴파일 캐시를 포함함
torch.compile() 과 통합해 그래프 모드 실행을...