Machine Learning Engineer, Runtime & Optimization

We are looking for engineers with ML software or ML systems expertise to help us improve compute performance on both cloud and car. You'll work across the entire ML stack from the system perspective, from efficient deep learning models, model compression, ML software (e.g. JAX, XLA, Triton, and CUDA), to.

Requirements

• M.S. in CS, EE, Deep Learning or a related field
• 2+ years of experience as a technical lead
• 5+ years of experience developing solutions in ML systems or ML software stack
• Deep understanding of ML system architecture, performance analysis and tools
• Strong Python or C++ programming skills

Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program