Lead Software Engineer - Kernels
MatX · Mountain View, CA · 4 wk ago
HybridEngineering$120k–$275k/yrFull-time
What You'll Do
- 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
Who You Are
- 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
- 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
Bonus Points
- If You Have Experience implementing kernels for ML models such as Transformers
- If You Have Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism
- If You Have Familiarity with how compilers work
What We Offer
- A Stake in our success
- A flexible cash equity compensation mix that fits your needs
- Health & Wellness
- Time To Recharge
- Support to Parents
- Learning & Development
- Financial Wellbeing
- Remote Perks
- MatX E[x]tras
Job Details
- The US base salary for this full-time position is determined based on a variety of factors including role, experience, location, job related skills, and relevant education and training.
- Career length is only a guideline for compensation.