Member of Technical Staff, ML

MatX•Mountain View, CA

2d•$160,000 - $800,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. Its mission is to make advanced AI models run as efficiently as physics allows. This role focuses on systems-oriented ML research, including training and optimizing large language models for MatX hardware, evaluating model quality, and building distributed infrastructure for training and inference. The position also contributes an ML perspective to hardware architecture and supports hardware/ML co-design through experimentation and RTL prototypes, working with other teams to validate numerics. Candidates must be authorized to work in the United States and work from MatX's Mountain View offices Tuesday through Thursday. The role requires access to information subject to U.S. export controls; employment is contingent on being able to perform the job in compliance with those laws without obtaining an export-control license. This is a full-time position.

Requirements

Strong research track record in LLMs, demonstrated through publications and/or shipped, open-sourced research.
PhD or equivalent experience.
Depth in at least one of reinforcement learning, quantization, or attention/architecture, including long-context methods.
Excellent software engineering skills, including building and scaling distributed training/inference codebases.
Must be authorized to work in the United States.
Must work from MatX's Mountain View offices Tuesday through Thursday.
Must be able to perform job functions in compliance with U.S. export-control laws without obtaining a license from U.S. export-control authorities.

Nice To Haves

Related experience in inference/training optimization, ML–hardware co-design, or evaluations.

Responsibilities

Train and optimize LLMs for MatX hardware.
Run quality evaluations.
Build and set up distributed infrastructure for training and inference.
Advise on hardware architecture from an ML perspective.
Co-design hardware through ML experimentation and RTL prototypes.
Work with other teams to validate numerics.

Benefits

Cash/equity compensation mix and option for early exercise.
Company-subsidized medical, dental, vision, and life insurance.
Health Savings Account with company contribution.
Four weeks of accrued paid time off, 12 company holidays, and up to three weeks of remote/flexible work per year.
Up to 12 weeks of paid parental leave, 10 weeks of pregnancy disability leave, flexible return-to-work hours, and a $6,000 Family Fund.
$1,500 annual professional development budget.
Team lunches and dinners, quarterly off-sites, regular town halls, and onsite meals Monday through Friday.
401(k) and/or Roth IRA with a 5% company contribution.
Flexible spending accounts for medical, dental/vision, dependent care, parking, and transit expenses.
Company-paid commute options, including rideshare, public transportation reimbursement, or a Fasttrak card.
$50 monthly allowance for employee-selected perks.
Remote-work support, including home-tech setup, up to $1,000 for office setup purchases, and Wi-Fi expense reimbursement.
Cellular reimbursement of $35 per month and Wi-Fi reimbursement of $40 per month.
Long-term disability and travel insurance.
100% paid mental health benefit and Employee Assistance Program.
Family Fund may cover fertility, surrogacy, adoption, menopausal health, preservation, family counseling, and related needs.