구글, TPU용 PyTorch 소프트웨어 스택 'TorchTPU' 오픈소스 공개하며 자체 칩 생태계 확장
구글은 TPU에 익숙한 PyTorch 코드를 적용할 수 있는 'TorchTPU'를 오픈소스로 공개하며 자체 칩 생태계 확장에 나섰다. 이는 개발자들이 최소한의 코드 수정으로 TPU에서 모델 학습, 추론 및 배포를 지원받을 수 있게 함으로써 하드웨어 선택지를 넓히는 것을 목표로 한다. 또한, 대규모 분산 학습 프레임워크인 'TorchTitan'에 TPU 지원을 통합하고, 온디바이스 AI 추론 도구까지 공개했다.
(새 창에서 원문 열기)