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.ResponsibilitiesRL environments to teach frontier LLMs how to design hardware and write kernelsEvals to measure model performance across frontier vendors for MatX tasksDifficult hardware / software optimization puzzles that challenge even the latest frontier modelsBuild tools for making these puzzles easier for both LLMs and humansSkills0 - 2 years experience with systems programming using Rust / Python / HDLsLinux / Docker experience is a mustCurrently pursuing a degree in Electrical Engineering, Computer Science, Math, or a related fieldNon-coursework projects preferred; in HDL, RTL experience preferredExperience optimizing hardware for area / timing / powerML experience, especially with LLMsExperience with CLI coding agents: incl. Codex / Claude Code / Gemini CLIBenefitsA Stake in our success. A cash/equity mix that fits your needs and option to do early exerciseHealth & 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 yearSupport to Parents. Up to 12 weeks of paid parental leave, regardless of your path to parenthoodLearning & Development. $1,500 yearly towards your professional development e.g. conferences, courses, and other learning opportunitiesTeam Connection. Team Lunches, quarterly off-sites, and regular town hallsFinancial 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 expensesCommute 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 mostRemote Perks. We work remotely Monday & Friday, supported by home-tech setup, and remote wifi expense reimbursementCompany OverviewMatX 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.



Apply To This Job