A gigante das buscas Google anunciou nesta terça-feira o lançamento do TorchTPU, sua aguardada versão do framework PyTorch, agora disponível como projeto de código aberto. Com essa movimentação, o Google se junta à Amazon na tentativa de enfraquecer o controle absoluto da Nvidia sobre o setor de inteligência artificial, onde a maioria dos aplicativos é desenvolvida para rodar nativamente nos chips da empresa americana, conhecidos como GPUs.
O PyTorch é um conjunto de ferramentas de código aberto bastante popular entre desenvolvedores de aplicativos de inteligência artificial e startups, sendo utilizado para construir, treinar e executar modelos de IA. Segundo estimativas do setor, aproximadamente 85% dos aplicativos de inteligência artificial são criados utilizando essa plataforma. O framework foi originalmente desenvolvido pela Meta em 2016, muito antes do mundo perceber o impacto transformador dos grandes modelos de linguagem. Naquela época, as GPUs da Nvidia já eram o padrão para pesquisas em inteligência artificial, o que fez com que muitas funcionalidades do PyTorch fossem construídas especificamente para funcionar com esses chips.
Em 2022, a Meta transferiu a gestão do PyTorch para a Fundação Linux, uma organização sem fins lucrativos que administra diversos projetos de código aberto relevantes. A versão do Google, chamada TorchTPU, permite que aplicativos de inteligência artificial rodem nos chips de IA do Google, conhecidos como Unidades de Processamento Tensor. Além disso, facilita a migração de aplicativos PyTorch existentes para essas unidades e, consequentemente, para a plataforma Google Cloud. Os desenvolvedores podem realizar essa transferência com uma simples alteração de configuração, sem necessidade de reescrever seus aplicativos.
A Amazon fez um anúncio semelhante na segunda-feira, quando lançou sua própria versão do PyTorch, denominada TorchNeuron. Essa ferramenta permite que aplicativos de inteligência artificial sejam transferidos para o chip de IA desenvolvido internamente pela Amazon, chamado Trainium, e para a plataforma AWS, também sem necessidade de reescrever o código. A empresa de comércio eletrônico e computação em nuvem informou ainda que vai além nessa estratégia para enfraquecer o domínio da Nvidia sobre o PyTorch, submetendo código ao projeto principal que desacopla diversas funcionalidades dos chips da Nvidia, tornando a plataforma mais aberta a qualquer hardware por padrão.
No entanto, o Google pode estar superando a Amazon em alguns aspectos. A empresa afirma ter trabalhado diretamente com a Meta, Hugging Face, Inferact criadora do vLLM, RadixArk criadora do SGLang, entre outros, para garantir que os recursos e bibliotecas de código do TorchTPU funcionem bem com essas ferramentas populares de desenvolvimento de inteligência artificial, assim como o PyTorch original. Além disso, a gigante das buscas lançou uma ferramenta dedicada, chamada conjunto de agentes TPU, para ajudar desenvolvedores a migrarem seus aplicativos para os chips e nuvem do Google, corrigindo erros ao longo do processo. A empresa também atualizou sua ferramenta LiteRT, popular para executar inteligência artificial em celulares, laptops e dispositivos embutidos, para funcionar melhor com aplicativos de inteligência artificial desenvolvidos em PyTorch.
Vale ressaltar que nada disso significa que desenvolvedores de aplicativos de inteligência artificial abandonarão automaticamente o hardware da Nvidia em massa simplesmente porque agora podem migrar facilmente para chips e nuvens alternativas. Porém, Amazon e Google estão claramente trabalhando para acabar com a vantagem da Nvidia no mercado.
Fonte: TechCrunch
