Google lança TorchTPU de código aberto para facilitar a migração de apps PyTorch
🎧 Ouça esta notícia
O Google anunciou que o TorchTPU, sua versão do PyTorch, está disponível como projeto de código aberto. A ferramenta permite executar aplicativos de IA baseados em PyTorch nas unidades de processamento tensorial (TPUs) do Google e migrá-los para o Google Cloud com uma simples mudança de configuração, sem reescrever o código. O PyTorch é um kit de ferramentas de código aberto usado para criar, treinar e executar modelos de IA, enquanto muitos de seus recursos foram desenvolvidos para funcionar com GPUs da Nvidia.
A Amazon anunciou na segunda-feira o TorchNeuron, que oferece migração semelhante para o chip Trainium e a AWS, também sem reescrita. A Amazon ainda pretende contribuir com código ao projeto principal do PyTorch para tornar seus recursos menos dependentes do hardware da Nvidia. O Google diz ter trabalhado com a Meta, a Hugging Face, a Inferact e a RadixArk para compatibilizar o TorchTPU com ferramentas populares e lançou uma suíte de agentes para ajudar na migração e na correção de erros.
Por que importa · interpretação editorial
Ao abrir o TorchTPU e oferecer compatibilidade com ferramentas populares, o Google pode reduzir o trabalho de adaptação de aplicações PyTorch para seus TPUs e para o Google Cloud. A disputa com a iniciativa semelhante da Amazon também pode dar aos desenvolvedores mais alternativas de hardware sem reescrever o código, embora cada migração ainda dependa das ferramentas e integrações disponíveis.
Fontes
APIsHardware