Performance Modeling Engineer

• 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

• Strong background in performance modeling
• Strong working knowledge of latest LLM model architectures
• Experience with computer architecture, particularly AI accelerators

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

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.