AI Research Engineer

About The Role: We're seeking a Principal Machine Learning Lead 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. You'll work closely with our engineering team to design, implement, and iterate on large-scale AI models while building efficient systems for rapid experimentation and deployment. We have a small team of AI engineers, so we're looking for someone who is excited to work at a startup, and work across the stack to do what is needed to get a model that achieves a customer problem. We're looking for an experienced engineer ready to make a tangible impact in the AI robotics revolution. This role requires a proven background working on ML planning within the autonomous vehicle space with experience using the latest techniques in diffusion and autoregressive models in a professional setting. If you're excited about building the latest advancements in AI and working in the robotics space, we'd love to hear from you.

In this role, you will: 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
• You might thrive in this role if you:
• Have 5+ 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
• Have experience using the latest techniques in diffusion and autoregressive models in a professional setting
• Are familiar with training inference and infra pipelines that go from camera input to trajectory output for self driving or robotics
• Have experience with RL (reinforcement learning)
• Have a strong understanding of modern ML architectures and training techniques
• Have experience with model debugging, optimization, and performance tuning
• Have a background in implementing ML research papers and adapting academic work
• Tech StackPyTorch (experience with PyTorch required)
• PythonNodeJS/Typescript
• DockerCompensation and Benefits

Compensation Range: $225K - $300K