MatX is looking to develop and optimize AI models to run efficiently on their hardware, bringing advancements in AI quality and availability.
Requirements
- Excellent software engineering skills
- 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
- Authorized to work in the United States
- Work from our offices in Mountain View Tuesdays-Thursdays
- Capacity to perform job functions in compliance with U.S. export control laws without obtaining a license from U.S. export control authorities
