All vacancies
Senior/Lead AI/ML Performance Engineer
EPAM Systems
Remote · Czechia, SlovakiaSalary not disclosedfull-timeEPAM Systems Careers (remote abroad)
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
Languages
- Work format
- Remote
- Seniority
- Lead
- Posted
- 29 Sept 2026
- Last verified
- In the last 5 days
Similar roles
- Data Engineer (Snowflake)Addepto · Remote · worldwide not confirmed
- 1154 - Data AnalystGoFasti · Remote · worldwide, UTC+4 accepted
- Principal Streaming Technology & Quality LeadSoftConstruct · Remote · Armenia
- Rust - Senior Engineer - AI Interaction EvaluatorG2i · Remote · worldwide, UTC+4 accepted
- Product AnalystStellar Tech · Remote · worldwide, UTC+4 accepted
- Quantitative ResearcherEqvilent · Remote · worldwide