Software Engineer - Kernels

MatX•Mountain View, CA

9d•$160,000 - $600,000•Hybrid

About The Position

MatX is developing vertically integrated, full-stack solutions spanning silicon, systems, hardware, and software to train and run large machine-learning workloads. The company primarily uses Rust. The Software Engineer - Kernels will design and optimize kernels that interface directly with MatX hardware, partner with ML Research and Hardware Engineering teams, and provide programmer-focused expertise on hardware architecture to support seamless integration with the software stack. This is a hybrid position requiring work from the Mountain View, California office three days per week, Tuesday through Thursday.

Requirements

Bachelor’s degree in Computer Science or an equivalent degree.
10 or more years of software development experience.
Experience optimizing software for specialized hardware, using techniques such as parallelism, SIMD programming, C, assembly-level optimization, or GPU/CUDA programming.
Proficiency in at least one of the following languages: assembly, C++, C, Zig, or Rust.

Nice To Haves

Experience implementing kernels for machine-learning models such as Transformers.
Experience using and implementing distributed parallelism techniques such as AllReduce, AllToAll, data parallelism, or tensor parallelism.
Familiarity with how compilers work.

Responsibilities

Design and optimize kernels that interface directly with MatX hardware.
Work in partnership with 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.

Benefits

Four weeks of accrued paid time off, 12 company holidays, and up to three weeks of remote work.
Company-subsidized medical insurance for employees and dependents, Guardian dental and vision insurance for employees and dependents, employee life insurance, and HSA and FSA offerings through Lively.
Roth or pre-tax 401(k), or both, with a 5% company contribution regardless of employee contribution.
100% company-paid life insurance up to $300,000, long-term disability insurance, and travel insurance.
$1,500 annual professional development budget.
Onsite team lunch and dinner Monday through Friday.
100% coverage of daily commuting costs through a company rideshare account, public transportation reimbursement, or a FastTrak card.
$50 monthly allowance for a selected perk.
Monthly cellular reimbursement of $35 and Wi-Fi reimbursement of $40.
Up to $1,000 in office setup purchases through the company Amazon business account.
100% paid mental health benefit through SpringHealth and Guardian EAP.
Up to 12 weeks of paid parental leave regardless of path to parenthood, 10 weeks of pregnancy disability leave, flexible return-to-work hours, and a $6,000 Family Fund for eligible family-related needs.