Senior AI/ML Performance Engineer - TPU Benchmarking
Listed on 2026-10-06
-
Engineering
AI Engineer (Applied/Software)
EPAM Systems is seeking a Senior/Lead AI/ML Performance Engineer to benchmark and profile AI/ML workloads on TPU hardware, generating empirical data for the optimization solver's cost model. You will conduct JAX/XLA benchmarking on TPU slices and analyze TPU profiler traces to build hardware coefficient matrices.
The role involves profiling LLM training and inference performance, collaborating with optimization engineers to calibrate solver inputs and improve accuracy and efficiency of cost
We have an opening for a Senior AI/ML Performance Engineer - TPU Benchmarking in United States within IT & Technology, Engineering.
We invite applications for the Senior AI/ML Performance Engineer - TPU Benchmarking position located in United States.
The following role is for a Senior AI/ML Performance Engineer - TPU Benchmarking with EPAM Systems Inc.
Our group is growing, and we are hiring a Senior AI/ML Performance Engineer - TPU Benchmarking in United States.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).