핵심 요약
구글이 vLLM과 SGLang 인퍼런스 엔진을 TPU에서 PyTorch 백엔드로 제공하고, TorchTPU를 활용해 PyTorch 도구로 서빙 워크로드를 실행 및 최적화하는 솔루션을 발표했습니다.
구현 방법
- Native PyTorch 백엔드로 vLLM/SGLang 인퍼런스 엔진 제공
- TorchTPU 기반 TPU 커널 최적화 및 KV 캐시 관리(TPU Sync 포함)와 디스aggregated 서빙 도입
- 오픈 소스 Maintainers와의 협업으로 프로덕션급 기능 구현
주요 결과
- Ironwood TPU에서 장-context 및 에이전트형 워크로드에 대해 업계 선도적 성능 및 비용 효율성을 입증



