Lead C++ Developer

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 advanced toolchains that bridge them. Your work will directly enable AI research, production deployments and the broader open-source ecosystem, addressing complex technical issues that directly impact the efficiency and scalability of AI across the industry.

Responsibilities

• Design and optimize high-performance kernels (using languages like Pallas, Mosaic and Triton) targeting TPU and GPU architectures for critical Machine Learning (ML) operations, from massive training runs to high-speed inference.
• Architect infrastructure such as benchmarking suites, autotuning frameworks, performance analysis tools, regression testing and documentation to support custom kernels in key Open-Source Software (OSS) libraries.
• Track advancements in hardware architectures, compiler technologies and AI models to identify new opportunities for performance optimization through custom kernels.
• Engage with ML researchers, framework developers (JAX, PyTorch) and compiler engineers (XLA) to enhance adoption, identify new requirements and address bottlenecks.

Requirements

• Bachelor's degree or equivalent practical experience.
• 7+ years of industry experience.
• 5+ years of software development experience in C++ or Python.
• 3+ years of experience testing, maintaining or launching software products.
• 1+ year of experience in software design and architecture.
• Expertise in performance optimization at the kernel level.
• English proficiency at B2 level or higher.

Nice to have

• Experience optimizing TPU/GPU code using low-level kernel languages like Pallas, CUDA or Triton.
• Knowledge of ML frameworks (JAX, PyTorch), including attention, Mixture of Experts (MoEs), model optimization and low-precision formats.
• Understanding of modern accelerators, including data movement, pipelining, heterogeneous compute and scale-out.
• Understanding of compiler principles (optimization, code generation) and toolchains such as MLIR and OpenXLA.
• Experience building developer infrastructure, including OSS libraries, high-performance APIs and documentation.

We offer

• Engineering community of industry professionals and a friendly working environment.
• Flexible schedule and opportunity to work remotely within Poland, with the chance to work abroad for up to 60 days annually.
• Career roadmap, leadership development, certifications (GCP, Azure, AWS), and access to learning platforms.
• Stable income (Employment Contract or B2B), Employee Stock Purchase Plan and benefits package (health insurance, multisport, shopping vouchers).
• Modern offices with entertainment and relaxation zones, referral bonuses and corporate events.

Please note: The set of bonuses might vary based on the role you apply for – specifics will be discussed during the general interview. We will reach out to selected candidates exclusively.