Lead C++ Developer

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

Nice to have:

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

• Design and optimize high-performance kernels using languages like Pallas, Mosaic, and Triton targeting TPU and GPU architectures for critical ML operations.
• Architect infrastructure such as benchmarking suites, autotuning frameworks, performance analysis tools, regression testing, and documentation.
• Track advancements in hardware architectures, compiler technologies, and AI models to identify new performance optimization opportunities.
• Engage with ML researchers, framework developers (JAX, PyTorch), and compiler engineers (XLA) to enhance adoption, identify requirements, and address bottlenecks.