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
MatX is an AI semiconductor company that designs custom chips and hardware architectures to support large language models. It was founded in 2022, and is headquartered in Mountain View, California, USA, with a workforce of 11-50 employees. Its website is https://matx.com.