Google은 TorchTPU라는 네이티브 PyTorch 백엔드를 사용해 오픈소스 프로젝트 vLLM과 SGLang을 TPU에서 실행하는 새로운 추론 엔진을 발표했다. 이 엔진은 Inferact(vLLM), RadixArk(SGLang)와 공동 개발했으며, 이미 Meta 같은 기업의 프로덕션 워크로드를 처리하고 있다. 맞춤형 Pallas 커널과 KV 캐시 관리 등 TPU에 특화된 최적화를 적용해, PyTorch 사용자도 대규모·저지연 추론 서비스를 더 쉽게 운영하도록 하는 것이 목표다.
왜 중요한가
PyTorch 개발자는 이제 JAX로 전환하지 않고도 Google TPU 하드웨어에서 대규모 언어 모델의 빠르고 비용 효율적인 추론을 배포할 수 있다.