구글, TPU용 PyTorch 백엔드 오픈소스 공개
원제: Google open-sources TorchTPU, a native PyTorch back end for TPUs
왜 중요한가
PyTorch 생태계에서 TPU 접근성이 높아지면, NVIDIA GPU 의존도를 낮추려는 기업들의 선택지가 실질적으로 넓어진다.
구글이 2025년 TPU 전용 네이티브 PyTorch 백엔드 라이브러리 'TorchTPU'를 GitHub에 오픈소스로 공개했다. 이 프로젝트는 PyTorch 모델을 TPU에서 직접 실행할 수 있도록 커스텀 ATen 커널, 컴파일 캐시, 분산 학습 예제를 포함하며, 현재 Star 398개, Fork 20개를 기록 중이다.
구글이 'torch_tpu' 저장소를 통해 TPU 전용 PyTorch 백엔드를 오픈소스로 공개했다. 기존에 PyTorch로 TPU를 활용하려면 XLA(Accelerated Linear Algebra) 컴파일러 레이어를 거쳐야 했으나, TorchTPU는 네이티브 PyTorch 백엔드 방식으로 TPU에 직접 연결되는 구조를 채택했다.
주요 구성 요소는 세 가지다. 첫째, PyTorch 연산을 TPU에서 처리하기 위한 커스텀 ATen 커널. 둘째, 반복 실행 시 성능을 높이는 컴파일 캐시. 셋째, 분산 학습과 모델 병렬화를 위한 예제 코드다. 빌드 시스템으로는 Bazel을 사용하며, Docker 환경과 CI 파이프라인도 함께 제공된다.
저장소에는 4,466개의 커밋이 쌓여 있어 상당 기간 내부 개발이 진행된 것으로 보인다. 48개의 열린 이슈와 221개의 풀 리퀘스트가 존재하며, 현재도 활발히 개발 중임을 보여준다. 문서는 ReadTheDocs를 통해 별도 제공된다.
구글은 CLAUDE.md와 AGENTS.md 파일도 저장소에 포함시켜 AI 에이전트 기반 코드 기여 워크플로도 지원하고 있다.