Logo for EPAM Systems

Senior/Lead AI/ML Performance Engineer

EPAM Systems

Role overview

Qualifications

  • 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 TensorFlow, PyTorch, and LangChain

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)

Key facts

Hard skills

Other skills

  • Collaboration

About the company

EPAM Systems logo

EPAM Systems

IT Services & IT Consulting

Since 1993, EPAM Systems, Inc. (NYSE: EPAM) has used its software engineering expertise to become a leading global provider of digital engineering, cloud and AI-enabled transformation services, and a leading business and experience consulting partner for global enterprises and ambitious startups. We address our clients’ transformation challenges by fusing EPAM Continuum’s integrated strategy, experience and technology consulting with our 30+ years of engineering execution to speed our clients’ time to market and drive greater value from their innovations and digital investments. We make GenAI real with our AI LLM orchestration, testing and engineering solutions, EPAM DIAL, EPAM EliteA™ and EPAM AI/RUN™, respectively. We deliver globally, but engage locally with our expert teams of consultants, architects, designers and engineers, making the future real for our clients, our partners and our people around the world. We believe the right solutions are the ones that improve people’s lives and fuel competitive advantage for our clients across diverse industries. Our thinking comes to life in the experiences, products and platforms we design and bring to market. Added to the S&P 500 and the Forbes Global 2000 in 2021 and recognized by Glassdoor and Newsweek as a Top 100 Best Workplace, our multidisciplinary teams serve customers across six continents. We are proud to be among the top 15 companies in Information Technology Services in the Fortune 1000 and to be recognized as a leader in the IDC MarketScapes for Worldwide Experience Build Services, Worldwide Experience Design Services and Worldwide Software Engineering Services. Learn more at www.epam.com.

Company details

Company typeXLarge
IndustryIT Services & IT Consulting
Company size10001

Links

Website LinkedIn See all jobs

Your match analysis

See how your profile stacks up against this role.

We compared the job requirements to your profile to show where you're strong and where you fall short.

Job description

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 Requirements 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 TensorFlow, PyTorch, and LangChain Experience with JAX/XLA and TPU-specific profiling tools Nice to have Experience with GPU/TPU performance benchmarking at scale Background in ML systems or MLPerf-style benchmarking

Apply once. Then go straight to the hiring manager.

After you apply, unlock the direct contact details of the people who actually make the call. A quick follow-up makes you 5x more likely to land an interview.

MR

Marcus Rivera

Chief Revenue Officer

m.rivera@company.com
linkedin.com/in/marcusrivera
Unlocked after you apply
·

AI/ML Engineer Related jobs

Europe

Other jobs at EPAM Systems

Premium

Reach out to the hiring manager directly.

Gain access to the contact details of the hiring managers who actually decide, and reach out to network with them directly. That, plus more when you upgrade:

  • Full match report with fit score and gaps
  • Career diagnostics on how recruiters read you
  • Curated company matches and warm intros
  • 48h early access to new roles

Cancel anytime.