MatX
Lead Software Engineer - Kernels
Mountain View, CA · Contract
No sponsorship$120k-$275kDetected 7 days ago
C++Full-Stack DevelopmentMachine LearningNLPHardware DesignResearchCommunicationCollaboration
About the role
- At the same time, you will be an active coding contributor by designing and implementing high-performance compute kernels for specialized AI hardware.
- This position requires access to information that is subject to U.S. export controls.
- MatX does not accept unsolicited resumes from individual recruiters or third-party recruiting agencies in response to job postings.
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
- Ability to navigate ambiguity, manage escalations, and align stakeholders around infrastructure decisions.
- Commute On Us For those commuting up to 1 hour, put your rideshare cost on our company card and reclaim the drive-time to get work done!
Requirements
- 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
- Excellent communication and collaboration skills, with a track record of translating complex technical trade-offs with the founders and stakeholders
- Experience implementing kernels for ML models such as Transformers
- Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.
- Familiarity with how compilers work
Compensation
- Early Career - $120,000 - $275,000 + equity
- Mid Career - $175,000 - $400,000 + equity
- Senior Career - $250,000 - $600,000 + equity
- A Stake in our success A flexible cash equity compensation mix that fits your needs
- Time To Recharge 4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per year
Benefits
- A Stake in our success A flexible cash equity compensation mix that fits your needs
- Health & Wellness Company subsidized Health, Dental, Vision, and Life insurance; Pre-tax Health Savings Accounts with generous company contribution (even if you don't)
- Support to Parents Up to 12 weeks of paid parental leave, regardless of your path to parenthood
- Learning & Development $1,500 yearly towards your professional development e.g. conferences, courses, and other learning opportunities
- Financial Wellbeing 401K and/or Roth IRA, with 5% company contribution, even if you don't!
- Flexible Spending Accounts Pre-tax spend accounts for medical, dental/vision, dependent care, parking, and transit expenses
- MatX E[x]tras $50 per month to use on the perks you care about most
- Time To Recharge 4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per year
- Remote Perks We work remotely Monday & Friday, supported by home-tech setup, and remote wifi expense reimbursement
Visa & Work Authorization
- All candidates must be authorized to work in the United States and work from our offices in Mountain View Tuesdays-Thursdays.
This listing is sourced directly from MatX's careers page and normalized into a canonical job model.