AI Tooling Engineer
MatX is on a mission to enhance AI model efficiency and is developing full-stack solutions for large ML workloads. They are seeking an AI Tooling Engineer to work on RL environments, model performance evaluation, and 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
- 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
- Health & Wellness. Company subsidized Health, Dental, Vision, and Life insurance; Pre-tax Health Savings Accounts with generous company contribution (even if you don’t)
- Time To Recharge. 4 weeks paid time off (accrued), 12 company holidays, and 3 weeks remote/flexible work per year
- Support to Parents. Up to 12 weeks of paid parental leave, regardless of your path to parenthood
- Learning & Development. $1,500 yearly towards your professional development e.g. conferences, courses, and other learning opportunities
- Team Connection. Team Lunches, quarterly off-sites, and regular town halls
- Financial Wellbeing. 401K and/or Roth IRA, with 5% company contribution, even if you don’t!
- Flexible Spending Accounts. Pre-tax spend accounts for medical, dental/vision, dependent care, parking, and transit expenses
- Commute On Us. For those commuting up to 1 hour, put your rideshare cost on our company card and reclaim the drive-time to get work done!
- MatX E[x]tras. $50 per month to use on the perks you care about most
- Remote Perks. We work remotely Monday & Friday, supported by home-tech setup, and remote wifi expense reimbursement
Company Overview