Math Kernel Engineer

We’re seeking a Math Kernel Engineer to design and build the critical software layer that transforms high-level operations into optimized low-level linear algebra primitives. You'll work at the intersection of hardware and software, ensuring our systems achieve maximum performance.

• Design and implement kernel tooling
  :
  Architect and implement systems that decompose high-level software operations into efficient linear algebra primitives
• Performance analysis and optimization:
  Profile workloads to identify bottlenecks and drive informed decisions across hardware and software teams
• Cross-functional technical leadership:
  Establish and drive consensus on hardware-software interfaces, ensuring seamless integration between teams

• 5+ years of relevant software engineering experience
• BS or MS in Computer Science, Mathematics, Engineering, or related field (OR significant open-source contributions to related projects)
• Strong foundation in linear algebra and numerical computing
• Experience with performance profiling and optimization

• Bare-metal or RTOS development experience
• Deep familiarity with AI framework internals (PyTorch, Tensor Flow, JAX)
• Experience with scientific computing libraries (LAPACK, BLAS, MKL, cuBLAS)

• Mid-Senior level

• Full-time

• Software Development and Computer Hardware Manufacturing

Location:

Mountain View, CA

Salary range: $–$