Performance Modeling Engineer

Employer description

What MatX is BuildingMatX’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.

What You’ll Do Here:Translate AI accelerator architecture into performance models

Maintain modeling framework capabilities and releases

Automate creation of detailed tensor layouts for frontier LLMs

Create tools, compilers, and automations to integrate with the broader organization tech stack

Collaborate with ML, Software, and Architecture teams to support experiments and workflows

Who You Are:Strong background in performance modeling

Strong working knowledge of latest LLM model architectures

Experience with computer architecture, particularly AI accelerators

CompensationThe 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 - $120,000 - $275,000 + equity

Mid Career - $175,000 - $450,000 + equity

Senior Career - $275,000 - $600,000 + equity

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 Wellbeing: Choose from Roth IRA or 401K (or both) retirement plans with up to 5% company contribution to 401K (even if you don’t contribute). Also, 100% company-paid life insurance (up to $300K) and long-term disability insurances.

Professional Development: $1500 Professional Development Budget (per year)

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: Commute on our company Uber account, or reimburse your train rides. Either way, we pay 100% for your daily commute.

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 Benepass reproductive health & parental benefit.

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.

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.