Skip to content
ITA Jobs
All vacancies
EPAM Systems logo

Lead C++ Developer

EPAM Systems

Remote · PolandSalary not disclosedfull-timeEPAM Systems Careers (remote abroad)
Apply on company siteMatch my CV

We are seeking a talented Lead C++ Developer to join the highly interdisciplinary CoreML team, where you will drive the performance and optimization of both training and serving, delivering massive impact for customers. In this role, you will have exposure to the newest Tensor Processing Unit (TPU), Graphics Processing Unit (GPU) hardware, the latest ML models, and the advanced toolchains that bridge them.

Responsibilities

  • Design and optimize high-performance kernels (using languages like Pallas, Mosaic and Triton) targeting Tensor Processing Unit (TPU) and Graphics Processing Unit (GPU) architectures for critical Machine Learning (ML) operations, redefining what's possible from massive training runs to high-speed inference
  • Architecture of infrastructure such as benchmarking suites, autotuning frameworks, performance analysis tools, regression testing and documentation, transforming how the developer community interacts with increasingly critical custom kernels in key Open-Source Software (OSS) libraries
  • Track the latest advancements in hardware architectures, compiler technologies and AI models to identify new opportunities for performance optimization through custom kernels
  • Engagement with ML researchers, framework developers (Just After eXecution (JAX), PyTorch) and compiler engineers (Accelerated Linear Algebra (XLA)) to enhance adoption, identify new requirements and address bottlenecks by providing appropriate solutions

Languages

    Work format
    Remote
    Seniority
    Lead
    Posted
    29 Sept 2026
    Last verified
    In the last 5 days

    Similar roles

    Keep looking

    More Software jobs in Armenia and remoteAll IT vacanciesInternships and entry level