AI Tooling Engineer

MatX is focused on developing cutting-edge AI solutions that integrate hardware and software for optimal performance. The AI Tooling Engineer will work on creating reinforcement learning environments, evaluating model performance, and solving complex hardware/software optimization challenges.

Responsibilities

• RL environments to teach frontier LLMs how to design hardware and write kernels
• Evals to measure model performance across frontier vendors for MatX tasks
• Difficult hardware / software optimization puzzles that challenge even the latest frontier models
• Build tools for making these puzzles easier for both LLMs and humans

Skills

• 0 - 2 years experience with systems programming using Rust / Python / HDLs
• Linux / Docker experience is a must
• Currently pursuing a degree in Electrical Engineering, Computer Science, Math, or a related field
• 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
• Non-coursework projects preferred; in HDL, RTL experience preferred
• Experience optimizing hardware for area / timing / power
• ML experience, especially with LLMs
• Experience with CLI coding agents: incl. Codex / Claude Code / Gemini CLI

Benefits

• A Stake in our success. A cash/equity mix that fits your needs and option to do early exercise
• Company subsidized Health, Dental, Vision, and Life insurance; Pre-tax Health Savings Accounts with generous company contribution (even if you don’t)
• 4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per year
• Up to 12 weeks of paid parental leave, regardless of your path to parenthood
• $1,500 yearly towards your professional development e.g. conferences, courses, and other learning opportunities
• Team Lunches, quarterly off-sites, and regular town halls
• 401K and/or Roth IRA, with 5% company contribution, even if you don’t!
• Pre-tax spend accounts for medical, dental/vision, dependent care, parking, and transit expenses
• For those commuting up to 1 hour, put your rideshare cost on our company card and reclaim the drive-time to get work done!
• $50 per month to use on the perks you care about most
• We work remotely Monday & Friday, supported by home-tech setup, and remote wifi expense reimbursement

Company Overview