Machine Learning Infrastructure Engineer
Posted Updated
Description
- Provide infrastructure support to our ML research and product
- Build tooling to diagnose cluster issues and hardware failures
- Monitor deployments, manage experiments, and generally support our research
- Maximize GPU allocation and utilization for both serving and training
Benefits
- 💰 We invest in your future with a generous 401(K) contribution
- Top-notch health coverage for you & your family, with majority of the premium covered
- 🍼 New parents, we’ve got you covered with incredible paid leave -up to 20 weeks
- 🌴 4 weeks of PTO to explore, unwind & come back recharged
- 🍽️ Daily in-office catering plus a monthly Doordash stipend to help keep you fueled no matter where you are
- ✨ Monthly wellness stipend to support you in your health journey
- Experience with cloud platforms (e.g., Compute Engine, Kubernetes, Cloud Storage)
- Experience with supporting large language model training
- Experience with ML frameworks like Pytorch/TensorFlow/JAX
- Experience with GPU kernel development
- Experience working with GPUs
- Experience with large GPU clusters and high-performance computing/networking
- Experience in developing tools used to diagnose ML infrastructure problems and failures
- 4+ years of experience supporting the infrastructure within an ML environment
- We’re looking for seasoned ML Infrastructure engineers with experience designing, building and maintaining training and serving infrastructure for ML research