Lead Software Engineer, Kernels at Matx · LumiJobs
senior
$160k–$275k
H1B: yes
Matx Lead the Kernel team at MatX by managing engineers and designing high-performance compute kernels for specialized AI hardware. Collaborate with ML and hardware teams to co-design next-generation AI architectures.
Responsibilities
Manage, mentor, and build a team of highly specialized kernel engineers and software developers
Design and optimize kernels that interface directly with our hardware
Work in partnership with our ML Research and Hardware Engineering teams
Provide expertise and guidance on hardware architecture from a programmer's perspective, ensuring seamless integration with the software stack
Qualification
LanguageAbility to navigate ambiguityFamiliarity with how compilers work
Required
2+ years of engineering team management experience
7+ years of working directly within engineering teams experience
Bachelor of Computer Science or equivalent degree
Experience optimizing software for specialized hardware, employing techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programming
Language: at least one of assembly, C++, C, Zig, or Rust
Ability to navigate ambiguity, manage escalations, and align stakeholders around infrastructure decisions.
Excellent communication and collaboration skills, with a track record of translating complex technical trade-offs with the founders and stakeholders
Preferred
Experience implementing kernels for ML models such as Transformers
Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.
