구글, 파이토치 기본 하드웨어에 TPU 올렸다…GPU·CPU와 나란히

토큰포스트 속보 2026.10.21 10:46

구글이 메타 파이토치 팀과 공동 개발한 오픈소스 TorchTPU를 공개했다. TPU는 GPU·CPU와 함께 파이토치가 기본 지원하는 하드웨어 옵션이 됐다.

토포AI로 이 기사 더 깊이 읽기

구글($GOOGL)이 메타($META)의 파이토치(PyTorch) 팀과 공동 개발한 오픈소스 프로젝트 TorchTPU를 공개했다. 이에 따라 TPU는 GPU·CPU와 나란히 파이토치가 기본 지원하는 하드웨어 옵션이 됐다.

구글 클라우드(Google Cloud) 공식 블로그는 구글이 'PyTorch Conference 2026'에서 TorchTPU를 오픈소스로 내놨다고 밝혔다. 코드는 깃허브(GitHub)에 공개됐으며, 기존 파이토치 코드는 소폭 수정만으로 TPU로 옮길 수 있다.

연동 범위도 넓다. 구글 클라우드는 vLLM, SGLang, TorchTitan, VeRL, Ray 등 프레임워크가 이미 TorchTPU와 연결됐다고 설명했다. 허깅페이스(Hugging Face) 트랜스포머스 v5에서는 device_map="tpu"를 넘기면 모델을 TPU에 배포할 수 있다.