Lead Software Engineer - Kernels

What MatX Is BuildingMatX is on a mission to be the compute platform for AGI. We are developing vertically integrated full-stack solutions from silicon to systems including hardware and software to train and run the largest ML workloads for AGI. We are seeking a hands-on engineering TLM to lead the Kernel team. You will manage, mentor, and grow a team of kernel engineers while setting technical direction and executing. At the same time, you will be an active coding contributor by designing and implementing high-performance compute kernels for specialized AI hardware.You will partner with the ML team to maximize output and improve diagnostic tools, and collaborate with hardware to co-design next-generation AI architectures.What You'll Do HereManage, mentor, and build a team of highly specialized kernel engineers and software developersDesign and optimize kernels that interface directly with our hardwareWork in partnership with our ML Research and Hardware Engineering teamsProvide expertise and guidance on hardware architecture from a programmer's perspective, ensuring seamless integration with the software stackWho You Are2+ years of engineering team management experience7+ years of working directly within engineering teams experienceBachelor of Computer Science or equivalent degreeExperience optimizing software for specialized hardware, employing techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programmingLanguage: at least one of assembly, C++, C, Zig, or RustAbility 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 stakeholdersBonus Points If You HaveExperience implementing kernels for ML models such as TransformersExperience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.Familiarity with how compilers workCompensationThe 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.Early Career - $120,000 - $275,000 + equityMid Career - $175,000 - $400,000 + equitySenior Career - $250,000 - $600,000 + equityWhat We OfferA Stake in our success A flexible cash equity compensation mix that fits your needsHealth & Wellness Company subsidized Health, Dental, Vision, and Life insurance; Pre-tax Health Savings Accounts with generous company contribution (even if you don’t)Time To Recharge 4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per yearSupport to Parents Up to 12 weeks of paid parental leave, regardless of your path to parenthoodLearning & Development $1,500 yearly towards your professional development e.g. conferences, courses, and other learning opportunitiesTeam Connection Team Lunches, quarterly off-sites, and regular town hallsFinancial 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 expensesCommute 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!MatX E[x]tras $50 per month to use on the perks you care about most Remote Perks We work remotely Monday & Friday, supported by home-tech setup, and remote wifi expense reimbursementAs part of our dedication to the diversity of our team and our focus on creating an inviting and inclusive work experience, MatX is committed to a policy of Equal Employment Opportunity and will not discriminate against an applicant or employee on the basis of race, color, religion, creed, national origin or ancestry, sex, gender, gender identity, gender expression, sexual orientation, age, physical or mental disability, medical condition, marital/domestic partner status, military and veteran status, genetic information or any other legally recognized protected basis under federal, state or local laws, regulations or ordinances.All candidates must be authorized to work in the United States and work from our offices in Mountain View Tuesdays-Thursdays.This position requires access to information that is subject to U.S. export controls. This offer of employment is contingent upon the applicants capacity to perform job functions in compliance with U.S. export control laws without obtaining a license from U.S. export control authorities.MatX does not accept unsolicited resumes from individual recruiters or third-party recruiting agencies in response to job postings. No fee will be paid to third parties who submit unsolicited candidates directly to our hiring managers or People team and any resumes submitted are deemed to be the property of MatX.