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 to deliver major customer impact. The role provides exposure to the newest Tensor Processing Unit (TPU) and Graphics Processing Unit (GPU) hardware, the latest ML models, and advanced toolchains that bridge them. Your work will enable AI research, production deployments, and the broader open-source ecosystem by addressing complex issues affecting AI efficiency and scalability.

Responsibilities

• Design and optimize high-performance kernels using languages such as Pallas, Mosaic, and Triton for TPU and GPU architectures, supporting critical ML operations from large-scale training 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 advances in hardware architectures, compiler technologies, and AI models to identify custom-kernel optimization opportunities.
• Collaborate with ML researchers, framework developers working with JAX and PyTorch, and compiler engineers working with XLA to improve 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.
• 1+ year of experience with software design and architecture.
• LLM experience.
• Expertise in kernel-level performance optimization.
• English proficiency at B2 level or higher.

Nice to have

• Experience optimizing TPU/GPU code with 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 OSS libraries, flexible high-performance APIs, and accessible documentation.

We offer

• Hybrid-by-design work mode 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 and well-being programs.
• Certification opportunities: 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.

Benefits listed above are available to employees only. Contractor cooperation is available; terms of B2B agreements are agreed individually. Only selected candidates will be contacted.