Google e AMD: Juntos em Novas TPUs

Chips de IA Híbridos com CPUs Integradas para Aprendizado por Reforço

Google e AMD Unem Forças para a Próxima Geração de TPUs

O Google está supostamente colaborando com a AMD no desenvolvimento de seus chips de Inteligência Artificial (TPUs) de 10ª geração. Esta parceria inovadora pode marcar a primeira grande incursão da AMD em um projeto de ASIC de IA personalizado e sugere que o Google está explorando um novo tipo de TPU híbrido.

A Busca por Núcleos de CPU e Empacotamento Avançado

  • Analistas da SemiAnalysis indicam que o Google pode estar interessado nos núcleos de CPU da AMD, ideais para cargas de trabalho que exigem mais processamento geral.
  • A AMD possui forte propriedade intelectual em empacotamento avançado, como o SoIC, e experiência em designs de data center híbridos (CPU+acelerador), como o Instinct MI300A.
  • Esta expertise é crucial, pois o Google dificilmente precisaria da AMD para um design convencional de TPU.

TPUs Híbridas para Aprendizado por Reforço

A colaboração aponta para um foco em TPUs que combinam a tecnologia de aceleradores do Google com núcleos de CPU no mesmo pacote, otimizados para cargas de trabalho específicas.

  • O Aprendizado por Reforço (RL) e modelos de agentes exigem consideravelmente mais poder de CPU em torno das operações do acelerador, diferentemente do treinamento convencional de LLMs.
  • O Google já tem aumentado os recursos de CPU: seus sistemas TPU 8i (para inferência, raciocínio e RL) usam um Google Axion CPU para cada dois TPUs, uma melhoria em relação à proporção de 1 Xeon para 4 TPUs nas gerações anteriores.

O Futuro da IA: Uma Abordagem Mais Centrada na CPU?

A integração direta de núcleos de CPU no pacote do TPU pode melhorar o desempenho e reduzir o consumo de energia, diminuindo a distância entre o processamento geral e o tensor. Isso sugere um futuro da IA mais dependente de CPU do que se imaginava.

Embora os detalhes exatos do envolvimento da AMD ainda sejam especulativos, o importante é que o Google está considerando uma nova família de TPUs v10 com forte componente de CPU, especialmente para cargas de trabalho de RL e agentes, com a AMD fornecendo os blocos de construção essenciais.

Baseado no artigo de Tom’s Hardware