AI Tooling Engineer
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. We are developing vertically integrated full-stack solutions from silicon to systems, including hardware and software, to train and run the largest ML workloads for AGI. We are looking for people who are excited about systems-focused ML research. 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
Requirements
- 0 - 2 years experience with systems programming using Rust / Python / HDLs. Linux / Docker experience is a must. Non-coursework projects preferred; in HDL, RTL experience preferred.
- Currently pursuing a degree in Electrical Engineering, Computer Science, Math, or a related field
Nice To Haves
- Experience optimizing hardware for area / timing / power
- ML experience, especially with LLMs
- Experience with CLI coding agents: incl. Codex / Claude Code / Gemini CLI
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
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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Entry Level
Education Level
No Education Listed
