Google이 TPU의 고대역폭 메모리, 호스트 RAM, 스토리지, 네트워크 간에 키-값(KV) 캐시와 모델 가중치를 옮기는 오픈소스 C++ 라이브러리 TPU Sync를 공개했다. 회사는 이 라이브러리를 사용하면 매개변수가 235 billion~480 billion인 전문가 혼합(MoE) 모델에서 KV 캐시 전송 지연을 90% 이상 줄일 수 있다고 밝혔다.
왜 중요한가
개발자는 TPU에서 대규모 언어 모델의 상태를 훨씬 빠르게 갱신해 추론이나 미세 조정 중 멈춤 시간을 줄일 수 있다.