Drimva

MatX

Software Engineer - Kernels

$160K–$600K

Mountain View (HQ) or Remote · Hybrid · Full-time · Posted Oct 28, 2026

Experience
10+ years of software development experience.
Education
Bachelor of Computer Science or equivalent degree.
Visa sponsorship
Employment is contingent on being able to perform the job in compliance with U.S. export control laws without obtaining an export-control license.

About the job

Design and optimize hardware-facing kernels for ML workloads, collaborating with research and hardware teams to integrate software with hardware architecture.

What you'll do

  • Design and optimize kernels that interface directly with hardware.
  • Partner with ML Research and Hardware Engineering teams.
  • Advise on hardware architecture from a programmer’s perspective to support software-stack integration.

What they ask for

  • Bachelor of Computer Science or equivalent degree.
  • 10+ years of software development experience.
  • Experience optimizing software for specialized hardware using techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programming.
  • Experience with at least one of assembly, C++, C, Zig, or Rust.

Nice to have

  • Experience implementing kernels for ML models such as Transformers.
  • Experience with distributed parallelism techniques or familiarity with how compilers work.

Pay

$160,000 - $275,000 USD base salary; $175,000 - $400,000 USD base salary; $250,000 - $600,000 USD base salary

Tools

Rust, SIMD, C, assembly, GPU/CUDA, C++, Zig

AI focus

Develop and optimize kernels for ML workloads; experience with ML model kernels such as Transformers is a bonus.

Model-written summary · the original posting is authoritative

First seen Oct 29, 2026 · Last verified Nov 4, 2026 (UTC)