AI Tooling Engineer at MatX in Mountain View, CA

Summary: 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 Required Qualifications: - 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 Preferred Qualifications: - 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 Required Skills: Rust, Python, HDLs Important Skills: Linux, Docker, ML experience Nice-to-Have Skills: Electrical Engineering, Computer Science, CLI coding agents 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