Google lança TorchTPU para executar PyTorch em chips TPU

Google lançou o TorchTPU, um backend da biblioteca PyTorch, para executar código nativo em chips TPU com desempenho otimizado. Esta iniciativa é semelhante ao lançamento do TorchNeuron pela Amazon, que compila código PyTorch para ser executado em chips Trainium.

O TorchTPU resulta de uma colaboração entre o Google e a equipe do PyTorch da Meta, visando expandir o uso do PyTorch para novos chips, além dos tradicionais CPU e GPU. O Google também se uniu à Hugging Face para testar a execução de diversos modelos abertos no TorchTPU, assegurando que o desempenho é satisfatório.