MatX is looking to develop vertically integrated full-stack solutions to train and run the largest ML workloads for AGI efficiently.
Requirements
- Experience training and tweaking neural networks, ideally LLMs
- Experience optimizing neural networks for hardware efficiency, for example regarding FLOPs, memory bandwidth, communication bandwidth, precision, parallel layout, batch sizes
Responsibilities
- Train and optimize LLMs for our hardware
- Run quality evaluations
- Build and set up distributed infrastructure for training and inference
- Advise on the hardware architecture from an ML perspective
Other
- Bachelor of Computer Science or equivalent degree by date of employment
- Excellent software engineering skills
- Authorized to work in the United States
- Work from our offices in Mountain View Tuesdays-Thursdays
- Commuter stipend available for employees within 60 minutes of the office
- 401(k) with contribution matching
- Health, vision, and dental insurance
- Life and AD&D insurance
