Performance Modeling Engineer
MatX•Mountain View, CA
•$120,000 - $600,000•Hybrid
About The Position
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. This role involves translating AI accelerator architecture into performance models, maintaining the modeling framework, and automating the creation of detailed tensor layouts for frontier LLMs. The engineer will also create tools, compilers, and automations to integrate with the broader organization tech stack and collaborate with ML, Software, and Architecture teams to support experiments and workflows.
Requirements
- Strong background in performance modeling
- Strong working knowledge of latest LLM model architectures
- Experience with computer architecture, particularly AI accelerators
Responsibilities
- 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
Benefits
- 4 weeks PTO (accrued)
- 12 company Holidays
- up to 3 weeks remote work
- Company-subsidized Medical (Kaiser or Anthem) for employees & dependents
- Guardian Dental and Vision insurances for employee & dependents
- life insurance (employee only)
- HSA and FSA offerings via Lively
- Roth IRA or 401K (or both) retirement plans
- up to 5% company contribution to 401K
- 100% company-paid life insurance (up to $300K)
- long-term disability insurances
- $1500 Professional Development Budget (per year)
- Onsite team lunch & dinner Monday - Friday
- Company Uber account for commute
- Reimbursement for train rides
- $50/mo to use on the perk you value most
- $35/mo for cellular reimbursement
- $40/mo for wifi reimbursement
- Up to $1000 in office setup purchases via Amazon business account
- 100% paid mental health benefit via SpringHealth
- Guardian EAP
- Up to 12 weeks paid parental leave regardless of path to parenthood
- 10 weeks pregnancy disability leave
- flexible return-to-work hours
- Benepass reproductive health & parental benefit
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
Upload and Match Resume
What This Job Offers
Job Type
Full-time
Career Level
Entry Level
Education Level
No Education Listed
