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