Lead Software Engineer - Kernels

What MatX Is Building MatX 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

Here 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 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 Bonus Points If You Have Experience implementing kernels for ML models such as Transformers Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, tensor parallelism.

Compensation

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.

. Early Career - $120,000 - $275,000 + equity Mid Career - $175,000 - $400,000 + equity Senior Career - $250,000 - $600,000 + equity

g. conferences, courses, and other learning opportunities Team Connection Team Lunches, quarterly off-sites, and regular town halls 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 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!

What We Offer

A Stake in our success A flexible cash equity

Perks

you care about most Remote

We work remotely Monday Friday, supported by home-tech setup, and remote wifi expense reimbursement As 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 i