Treffer: Optimización y portabilidad de un Transformer Encoder utilizando JAX ; Optimization and Portability of a Transformer Encoder Using JAX ; Optimització i portabilitat d'un Transformer Encoder utilitzant JAX

[ES] Este proyecto explora el uso de JAX para optimizar el rendimiento computacional de un Transformer Encoder, enfocándose en la eficiencia de la inferencia en distintos tipos de hardware como CPUs, GPUs y TPUs. Aprovechando las capacidades de diferenciación automática y compilación just-in-time (JIT) de JAX, se implementa una versión altamente vectorizada y paralelizada del encoder, evaluando mejoras en velocidad respecto a implementaciones tradicionales. ; [EN] This project explores the use of JAX to optimize the computational performance of a Transformer Encoder, focusing on the efficiency of inference on different types of hardware such as CPUs, GPUs and TPUs. Leveraging the automatic differencing and just-in-time compilation (JIT) capabilities of JAX, a highly vectorized and parallelized version of the encoder is implemented, evaluating speed improvements over traditional implementations. ; [CA] Este projecte explora l'ús de JAX per a optimitzar el rendiment computacional d'un Transformer Encoder, enfocant-se en l'eficiència de la inferència en diferents tipus de maquinari com CPUs, GPUs i TPUs. Aprofitant les capacitats de diferenciació automàtica i compilació just-in-time (JIT) de JAX, s'implementa una versió altament vectoritzada i paral·lelitzada del encoder, avaluant millores en velocitat respecte a implementacions tradicionals. ; Siwinska, Patricia (2025). Optimización y portabilidad de un Transformer Encoder utilizando JAX. https://riunet.upv.es/handle/10251/224100 ; TFGM