Jose Augusto Lima and Calebe Bianchini. 2026. Performance Analysis of TPU and GPU Architectures Using PyTorch and JAX for AI and HPC. In Proceedings of the 7th Regional School on High Performance Computing of the Northeast Region, July 08, 2026, Recife/PE, Brasil. SBC, Porto Alegre, Brasil, 73-77. DOI: https://doi.org/10.5753/erad-ne.2026.26645.