More jobs:
Senior/Lead AI/ML Performance Engineer
Job in
San Antonio, Bexar County, Texas, 78201, USA
Listing for:
EPAM Systems
Full Time
position
Listed on 2026-11-06
Job specializations:
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, AI Reliability/ Performance Engineer
Job Description & How to Apply Below
We are looking for a Senior/Lead AI/ML Performance Engineer to benchmark and profile AI/ML workloads on TPU hardware to generate the empirical performance data that feeds the optimization solver's cost model.
Responsibilities- Conduct JAX/XLA benchmarking on physical TPU slices
- Capture and analyze TPU Profiler traces
- Build hardware coefficient matrices for use in the optimizer's cost model
- Profile LLM training/inference performance (FLOPS, memory access, token throughput)
- Collaborate with Optimization Engineers to calibrate solver inputs
- Strong understanding of the modern AI/ML landscape and LLM architectures
- Hands-on experience with LLM models/pipelines (training and inference)
- Proficiency in Python and/or Golang
- Familiarity with Tensor Flow, PyTorch, and Lang Chain
- Experience with JAX/XLA and TPU-specific profiling tools
- Experience with GPU/TPU performance benchmarking at scale
- Background in ML systems or MLPerf-style benchmarking
Consider building your career as a Senior/Lead AI/ML Performance Engineer at EPAM Systems.
Position Requirements
10+ Years work experienceTo View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(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).
(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).