O TorchNeuron permite que programadores executem código escrito com a PyTorch nos chips de IA Trainium da Amazon com uma única alteração de configuração e sem a necessidade de reescrever programas, de acordo com a unidade de chips Annapurna Labs da Amazon.
A Amazon, que planeja anunciar o software em uma conferência de desenvolvedores no Vale do Silício nesta segunda-feira, afirmou que a ferramenta será lançada publicamente como uma oferta gratuita e de código aberto.
O PyTorch, a biblioteca de código aberto apoiada pela Meta, é usado pela maioria dos programadores de IA. A Nvidia dedicou mais de uma década à sua ferramenta de software chamada Cuda, que faz com que o PyTorch funcione bem nos chips da Nvidia.
Outras empresas de computação em nuvem, como o Google, que também possui seus próprios chips de IA personalizados, estão trabalhando para atrair desenvolvedores da Nvidia com ferramentas melhores para a PyTorch.
Colin Brace, vice-presidente de software da Annapurna Labs, disse que a iniciativa visa reduzir o número de etapas que os desenvolvedores de IA precisam seguir para testar os chips desenvolvidos internamente pela Amazon.
“O objetivo tem sido atender aos clientes onde eles estão”, disse ele à Reuters em uma entrevista.
