Work

Análise de Desempenho entre Arquiteturas TPU e GPU utilizando PyTorch e JAX para IA e HPC

Year: 2026
Type: conference-paper
Abstract: Este trabalho apresenta uma análise técnica da arquitetura Tensor Processing Unit (TPU), comparando-a com aceleradores GPU tradicionais em tarefas de computação de alto desempenho. Além de microbenchmarks de multiplicação de matrizes densas via JAX e PyTorch sobre o compilador XLA, o estudo introduz uma avaliação da largura de banda de memória (HBM High Bandwidth Memory). Os resultados demonstram que a TPU supera a GPU de referência não apenas no tempo de execução, mas também na vazão de dados, consolidando se como uma infraestrutura eficiente para aplicações científicas intensivas em dados.
Language: Portuguese
Open Access status: bronze
OpenAlex ID: W7167214606
FWCI:
Cites: 4
Cited by:
Related to: 10