OpenTPU: Uma reimplementação de código aberto da TPU do Google

2025-05-28
OpenTPU: Uma reimplementação de código aberto da TPU do Google

O ArchLab da UC Santa Barbara lançou o OpenTPU, uma reimplementação de código aberto da Tensor Processing Unit (TPU) do Google. Baseado em detalhes do artigo de pesquisa do Google, mas sem uma especificação formal, o OpenTPU difere em vários detalhes de implementação do design do Google. Atualmente, suporta multiplicação de matrizes e funções de ativação ReLU/sigmoid, mas faltam recursos como convolução e pooling. Implementado usando PyRTL, o projeto inclui simuladores de hardware e funcionais, juntamente com um verificador para validar os resultados. Embora não haja dados de síntese concretos para um OpenTPU completo de 256x256 nesta versão alfa, sua natureza de código aberto oferece valiosas oportunidades de aprendizado e melhoria para os pesquisadores.

Hardware