Member of Technical Staff, ML

MatX; Mountain View (CA); Hybride; USD 160 000 - 800 000; Plein temps

Avantages offerts par ce poste

Equity

Health insurance

PTO 4 weeks

Remote work

Professional development

401K

Commuter benefits

Family leave

Résumé du poste

MatX in Mountain View, CA, is advancing systems-focused ML research to run the world's best AI models efficiently on specialized hardware.

The team trains and optimizes LLMs for our hardware, builds distributed training and inference infrastructure, and collaborates on hardware architecture from an ML perspective. A PhD or equivalent in ML or related fields is preferred, with strong software engineering skills and a track record of open research.

Qualifications

• PhD or equivalent experience with publications or shipped/open-sourced research in LLMs.
• Deep expertise in reinforcement learning, quantization or attention/architecture; inference/training optimization is a plus.
• Strong software engineering skills for building and scaling distributed training/inference codebases.

Responsabilités

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

Connaissances

LLM research

Software engineering

Distributed training

ML hardware co-design

Formation

PhD or equivalent

Outils

Python

C++

Distributed training frameworks

Description du poste

What MatX Is Building

MatX's mission is to make the world’s best AI models run as efficiently as allowed by physics, bringing the world years ahead in AI quality and availability. 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 are looking for people who are excited about systems-focused ML research.

What You'll Do Here

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

Who you Are

• A strong research track record in LLMs (PhD or equivalent experience), demonstrated through publications and/or shipped, open-sourced research. Depth in at least one of: reinforcement learning, quantization or attention/architecture (including long-context methods). Related experience in inference/training optimization, ML–hardware co-design, evals is a plus
• Excellent software engineering skills, including building and scaling distributed training/inference codebases

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.

• Early Career - $160,000 - $300,000 + equity
• Mid Career - $180,000 - $425,000 + equity
• Senior Career - $400,000 - $800,000 + equity

What We Offer

• A Stake in our successA cash/equity mix that fits your needs and option to do early exercise
• Health & 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

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.

What We Offer

• Time off: 4 weeks PTO (accrued) + 12 company Holidays + up to 3 weeks remote work
• Health: Company-subsidized Medical (Kaiser or Anthem) for employees & dependents, Guardian Dental and Vision insurances for employee & dependents, and life insurance (employee only), plus HSA and FSA offerings via Lively.
• Financial Security: Choose from Roth or pre-tax 401K (or both) with a 5% company contribution (even if you don’t contribute), 100% company-paid life insurance (up to $300K), long-term disability and travel insurance.
• Professional Development: $1500 per year Professional Development Budget
• Team Meals: MatX provides onsite team lunch & dinner Monday - Friday, with your choice of ordering via WeBox, Specialty’s or via our reimbursement system
• Commute on Us: We pay 100% for your daily commute. Commute via our company rideshare account, reimburse public transportation, or get a Fasttrak card (or all three)
• MatX E[x]tras: $50/mo to use on the perk you value most
• Cell & Internet Reimbursement: $35/mo for cellular and $40/mo for wifi
• Remote Home Setup: Use our Amazon business account for up to $1000 in office setup purchases.
• Mental Wellbeing: 100% paid mental health benefit via SpringHealth and Guardian EAP.
• Support to Parents: Up to 12 weeks paid parental leave regardless of path to parenthood, 10 weeks pregnancy disability leave, flexible return-to-work hours, and a $6K Family Fund to cover fertility, surrogacy, adoption, menopausal health, preservation, family counseling and more.

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.
---------

This position requires access to information that is subject to U.S. export controls. This offer of employment is contingent upon the applicant's capacity to perform job functions in compliance with U.S. export control laws without obtaining a license from U.S. export control authorities.
--------