Lead C++ Developer

We are seeking a talented Lead C++ Developer to join a highly interdisciplinary CoreML team, driving the performance and optimization of both training and serving. The role provides exposure to the newest Tensor Processing Unit (TPU) and Graphics Processing Unit (GPU) hardware, the latest ML models, and advanced toolchains. The work enables AI research, production deployments, and the broader open-source ecosystem by addressing complex technical issues affecting the efficiency and scalability of AI.

Responsibilities

• Design and optimize high-performance kernels using languages such as Pallas, Mosaic, and Triton for TPU and GPU architectures across critical ML operations, from massive training runs to high-speed inference.
• Architect infrastructure including benchmarking suites, autotuning frameworks, performance analysis tools, regression testing, and documentation for custom kernels in key open-source software libraries.
• Track advancements in hardware architectures, compiler technologies, and AI models to identify performance-optimization opportunities through custom kernels.
• Collaborate with ML researchers, framework developers (JAX, PyTorch), and compiler engineers (XLA) to increase adoption, identify requirements, and resolve 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, plus 1 year of software design and architecture experience.
• LLM experience.
• Expertise in kernel-level performance optimization.
• English proficiency at B2 level or higher.

Nice to have

• Experience optimizing TPU/GPU code using low-level kernel languages such as Pallas, CUDA, or Triton.
• Knowledge of ML frameworks including JAX and PyTorch; common operations such as attention and 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, including optimization and code generation, and toolchains such as MLIR and OpenXLA.
• Experience building developer infrastructure, including open-source libraries, flexible high-performance APIs, and accessible documentation.

We offer

• Hybrid-by-design work model and the opportunity to work remotely within Poland.
• Opportunity to work abroad for up to 60 days annually.
• Business-driven relocation opportunities.
• Career development programs, thought leadership, mentoring, soft-skills programs, and well-being programs.
• Certifications in Anthropic, Gemini, GCP, Azure, and AWS.
• English classes.
• Stable pay.
• Participation in the Employee Stock Purchase Plan with a 15% discount.
• Benefits package, including health insurance, multisport, and shopping vouchers.
• Referral bonuses up to $2,000.
• Offices with entertainment and relaxation zones, table tennis, football, free snacks, coffee, and more.
• Corporate, social, and well-being events.

Please note

• Listed benefits are available to employees only.
• Contractor engagement is available; B2B cooperation terms are agreed individually.
• Only selected candidates will be contacted.