MatX · 2 months ago
Software Engineer - Kernels
Mountain View, CA
Full-time
Hybrid
New Grad, Entry, Mid Level
$120K/yr - $200K/yr
0+ years expMatX is on a mission to be the compute platform for AGI, developing full-stack solutions for large ML workloads. The Software Engineer - Kernels will design and optimize kernels, collaborating with research and engineering teams to ensure integration with the software stack.
Artificial Intelligence (AI)HardwareManufacturingSemiconductor
No H1B
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
Qualification
Required
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
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
Preferred
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
Benefits
A Stake in our success A cash/equity mix that fits your needs and option to do early exercise
Heath & 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 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 e.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!
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 reimbursement
Company
MatX
MatX is an AI chip startup that designs chips that support large language models.
11-50 employees
Funding
Current Stage
Early StageTotal Funding
$125MKey Investors
Spark CapitalNat Friedman
2024-11-22Series A· $100M
2024-03-26Seed· $25M
Leadership Team
R
Reiner Pope
Founder and CEO
M
Mike Gunter
Founder and CTO
Recent News
2024-11-24
Company data provided by crunchbase