We're seeking a Lead Kernel Engineer/Architect to join our team in the Netherlands in a hybrid working mode. If you're passionate about pushing advanced hardware accelerators to their limits, this is your chance to shape the future of AI performance and scalability.

De Rol

In this position, you will drive the optimization of critical machine learning operations for large-scale training and inference, utilizing cutting-edge hardware such as TPUs and GPUs, advanced ML models, and performance toolchains. Your contributions will facilitate faster AI research and production deployments on cloud platforms and within open-source ecosystems. You will work closely with researchers, compiler engineers, and framework developers to deliver optimized, high-performance solutions that define modern AI computation.

Verantwoordelijkheden

• Design and optimize high-performance kernels for TPU and GPU architectures using low-level programming frameworks such as Pallas, Triton, or Mosaic.
• Build and maintain performance infrastructure, including benchmarking suites, autotuning systems, regression testing frameworks, and tooling.
• Collaborate with ML framework developers (e.g., JAX, PyTorch) and compiler teams (XLA/MLIR) to integrate custom kernels and mitigate performance bottlenecks.
• Track advancements in accelerator hardware, compiler technology, and AI model design to identify kernel-level optimization opportunities.
• Develop clear documentation, APIs, and supporting OSS components that enhance developer usability and adoption.
• Analyze and resolve complex performance issues affecting large-scale distributed training and inference systems.

Vereisten

• Bachelor’s degree or equivalent practical experience.
• 12+ years of industry experience in software engineering or systems programming.
• 5+ years of software development experience using C++ or Python.
• 3+ years of experience in testing, maintaining, or launching software products, with at least 1 year in software design or architecture.
• Hands-on experience in performance optimization at the kernel level for accelerators or high-performance systems.

Nice to Have

• Proficiency in low-level accelerator programming (CUDA, Triton, Pallas).
• Familiarity with ML frameworks such as JAX or PyTorch and optimization techniques for attention layers, Mixture of Experts (MoE), and precision tuning.
• Strong understanding of modern hardware accelerators, including pipelining, data movement, and heterogeneous compute.
• Knowledge of compiler principles and intermediate representations (e.g., MLIR, OpenXLA).
• Experience building OSS developer infrastructure, APIs, and performance-critical libraries.
• Excellent problem-solving skills and the ability to collaborate in cross-functional engineering environments.

Wat We Bieden

• 26 paid holiday days.
• Pension plan scheme.
• Disability insurance (WGA Shortfall insurance).
• Long-term disability insurance (WIA Top up insurance).
• EPAM Employee Stock Purchase Plan (ESPP).
• Commuting costs reimbursement.
• Laptop + corporate sim card + corporate mobile device (subject to certain eligibility requirements).
• Bike lease.
• Employee Assistance Program.
• Corporate Programs including Employee Referral Program with rewards.
• Learning and development opportunities including in-house training and coaching, professional certifications, and courses.

We look forward to your application and the opportunity to work together!