Member of Technical Staff, ML

MatX

. Train and optimize LLMs for MatX hardware .

Posted 10/26/2026full-timeMountain View • California • United StatesLead💰 $160,000 - $800,000 per yearWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in training and optimizing Large Language Models (LLMs) for specialized hardware, with a strong focus on distributed infrastructure and ML-hardware co-design. Proven ability to conduct quality evaluations and collaborate across teams to validate numerical results.

Highest-signal resume keywords

PhD Or Equivalent ExperienceLarge Language Models (LLMs)Distributed Training/InferenceML-Hardware Co-DesignReinforcement Learning

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

Training OptimizationQuality EvaluationsSoftware EngineeringReinforcement LearningQuantizationAttention/ArchitectureLong-Context MethodsDistributed InfrastructureML ExperimentationRTL Prototypes

Industry Keywords

U.S. Export Control LawsNumerics ValidationOpen-Sourced ResearchScaling CodebasesML-Hardware Collaboration

About the role

Key responsibilities & impact

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

Requirements

What you’ll need

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
Experience building and scaling distributed training/inference codebases
All candidates must be authorized to work in the United States
Must be able to work from the Mountain View offices Tuesday–Thursday
Must be able to perform job functions in compliance with U.S. export control laws without obtaining a license
Experience in inference/training optimization, ML–hardware co-design, or evaluations is a plus

Benefits

Comp & perks

Cash/equity mix and option to do early exercise
Company-subsidized medical, dental, vision, and life insurance
Pre-tax HSA with company contribution
4 weeks paid time off
12 company holidays
Up to 3 weeks remote/flexible work per year
Up to 12 weeks paid parental leave
$1,500 yearly professional development budget
Team lunches, dinners, quarterly off-sites, and regular town halls
Roth and/or pre-tax 401(k) with 5% company contribution
Flexible spending accounts for medical, dental/vision, dependent care, parking, and transit
Company-paid daily commute, rideshare, public transportation, and/or Fasttrak card
$50 monthly perks allowance
$35 monthly cellular reimbursement
$40 monthly Wi-Fi reimbursement
Up to $1,000 remote home-office setup purchases
100% paid mental health benefit
10 weeks pregnancy disability leave
Flexible return-to-work hours
$6,000 Family Fund for fertility, surrogacy, adoption, menopausal health, preservation, family counseling, and more