Staff AI Research Engineer

About the role

We're seeking a Staff AI Research Engineer to develop and optimize our AI models and training systems. This is an exciting opportunity to apply your deep ML expertise to cutting-edge AI/robotics applications.

Responsibilities

• Design and implement state of the art ML models and training pipelines
• Apply novel machine learning techniques to wide range of robotics applications
• Develop efficient data and training strategies
• Implement model evaluation frameworks and metrics tracking
• Lead model development and iteration with focus on:
• Rapid experimentation and prototyping of new model architectures
• Performance optimization and model debugging
• Transfer learning and fine-tuning strategies
• Build robust evaluation and debugging systems to:
• Analyze model behavior and failure modes
• Implement interpretability tools and visualization frameworks
• Track and improve model metrics
• Collaborate with engineering team to optimize training infrastructure and deployment

Requirements

• Have 7+ years of AI modeling experience, specifically within the self-driving car industry (or PhD with 3+ years of AI modeling experience in the self-driving car industry)
• Proven track record developing and deploying large-scale ML models
• Experience using the latest techniques in diffusion and autoregressive models in a professional setting
• Familiarity with training inference and infra pipelines that go from camera input to trajectory output for self driving or robotics
• Experience with RL (reinforcement learning)
• A strong understanding of modern ML architectures and training techniques
• Experience with model debugging, optimization, and performance tuning
• Background in implementing ML research papers and adapting academic work

Qualifications

• Experience with PyTorch (experience with PyTorch required)
• Proficiency in Python and NodeJS/Typescript
• Experience with Docker

Benefits

We offer a competitive compensation range of $250,000 to $300,000, depending on experience. We also provide:

• Paid time off
• Medical/dental/vision insurance
• Life insurance
• Disability insurance
• 401(k) plan

All Full-Time Employees are eligible for Employee Stock Options.