Hardwares

Elon Musk diz ter memorizado projeto do AI5, chip que será usado nos carros e robôs da Tesla

Tesla avança com chip AI5 para inferência de IA, prometendo alta performance, baixo consumo e custo reduzido em carros autônomos e no robô Optimus.
Imagem de: Elon Musk diz ter memorizado projeto do AI5, chip que será usado nos carros e robôs da Tesla

Elon Musk comentou sobre o avanço da Tesla no desenvolvimento do AI5, chip voltado à inferência de inteligência artificial, durante uma conversa com o gestor de fundos Ron Baron.

Segundo o CEO, o componente terá papel central nos próximos carros com direção autônoma da empresa e no robô humanoide Optimus.

A Tesla trabalha no próprio chip como parte de uma estratégia para competir com NVIDIA e AMD em tarefas de inferência de IA. Musk também criou o projeto Stargate, voltado à fabricação de chips usados por suas empresas.

AI5 será usado em carros e no Optimus

Em abril deste ano, Musk confirmou que a Tesla havia concluído o tape-out do AI5 e enviado o projeto à Samsung para produção.

O tape-out representa a etapa em que o desenho do chip é fechado e encaminhado à fabricante responsável pela fabricação física do componente. Na época, Musk comparou o desempenho projetado do AI5 às arquiteturas Hopper e Blackwell, da NVIDIA.

Segundo ele, uma única unidade do AI5 teria desempenho próximo ao Hopper, enquanto uma configuração com dois chips poderia chegar ao nível do Blackwell, com custo bem menor e consumo reduzido de energia.

Durante a conversa com Ron Baron, Musk explicou que a Tesla mantinha dois projetos separados: o supercomputador Dojo e o AI5.

Na avaliação do executivo, nenhum dos dois seguia um caminho capaz de colocar a empresa em uma posição competitiva contra a NVIDIA.

"O chip AI5 será um ótimo chip. Toda a Tesla depende dele. É o chip que estará nos nossos carros autônomos de próxima geração e também é essencial para o robô Optimus. Então, esse programa estava em uma situação ruim. Ele não estava avançando, é um projeto bastante ambicioso e realmente não estava em um caminho de sucesso. Também tínhamos o programa Dojo, que estava indo razoavelmente bem, mas não seguia um caminho competitivo contra a NVIDIA."

A decisão foi concentrar as equipes em um único projeto. O desenvolvimento do Dojo e do AI5 passou a ser tratado dentro do mesmo esforço, com prioridade para o chip destinado aos veículos e robôs da Tesla.

TESLA D1 CHIP
O chip D1 da Tesla em um slide de apresentação compartilhado em 19 de agosto de 2021. Imagem: Tesla Inc/YouTube

Musk prevê eficiência até três vezes maior que chips da NVIDIA

A Tesla continuará usando chips da NVIDIA no treinamento de modelos de inteligência artificial. Para inferência, porém, Musk considera o AI5 mais adequado às necessidades da empresa. A principal diferença estaria na relação entre desempenho e consumo de energia.

Segundo Musk, o AI5 foi projetado para entregar alta capacidade de processamento com baixo gasto energético, algo importante dentro de carros e robôs, onde consumo, temperatura e espaço físico são fatores importantes.

"Eu juntei os dois programas em apenas um para colocar todo mundo focado no chip AI5, que é essencial. Continuamos usando NVIDIA para treinamento, mas precisamos do chip AI5 para inferência. É um chip muito poderoso, mas também consome pouca energia. A eficiência por watt é extremamente boa. Provavelmente será pelo menos duas a três vezes melhor que a NVIDIA em desempenho por watt, no nível de inferência dentro do carro e do robô. E deve custar algo como 10% do preço de um chip da NVIDIA."

A comparação feita por Musk se refere especificamente ao desempenho por watt em tarefas de inferência realizadas dentro dos carros e do Optimus. Ele não colocou o AI5 como substituto dos aceleradores da NVIDIA usados para treinar grandes modelos de IA.

Projeto levou Musk a trabalhar também nos fins de semana

A importância atribuída ao AI5 também foi usada por Musk para explicar por que passou a dedicar longas jornadas de trabalho ao projeto, inclusive nos fins de semana.

O objetivo, segundo ele, era colocar o desenvolvimento do chip novamente dentro do cronograma necessário para os produtos futuros da Tesla.

"Esses números são muito importantes e eu precisava colocar o programa do chip de volta nos trilhos, então dediquei tempo a isso. Neste ponto, tenho todo o projeto físico do chip memorizado. Consigo visualizar tudo."

O chip AI5 da Tesla concentra agora boa parte dos planos da empresa para processamento local de inteligência artificial.

Se as metas citadas por Musk forem atingidas, o mesmo chip servirá de base para a próxima geração de sistemas de direção autônoma da empresa e para o Optimus, enquanto os chips da NVIDIA continuarão presentes na etapa de treinamento dos modelos.