Logo
MatX

Software Engineer - Kernels

MatX, Mountain View, California, us, 94039

Save Job

Overview

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 primarily use the Rust programming language. Responsibilities

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 Requirements

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 This is a hybrid role that will require you to work from our Mountain View, CA office 3 days a week on Tuesday through Thursday Preferences

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

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. Our offers allow candidates to swap cash for equity. The broad ranges for mid/senior candidates reflect greater flexibility for cash/equity swap. 0-5 years of experience - $120,000 - $200,000 + equity 5-10 years of experience - $120,000 -$300,000 + equity 10+ years experience - $120,000 - $400,000 + equity What We Offer

A Stake in our success

Generous equity, with option cash/equity swap at offer, and option to employee early exercise. Health & Wellness

Company subsidized Health, Dental, Vision, and Life insurance; Pre-tax Health Savings Accounts with generous company contribution Time To Recharge

4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per year 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 Team Connection

Team Lunches, quarterly off-sites, and regular town halls Financial Wellbeing

401K and/or Roth IRA, with 5% company contribution 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, reimburse ride-share costs and drive-time MatX E[x]tras

$50 per month to use on the perks you care about most Remote Perks

We work remotely Monday & Friday, with home-tech setup support and remote Wi-Fi 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 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.

#J-18808-Ljbffr