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. 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 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

• 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 Stack

• PyTorch (experience with PyTorch required)

• Python

• NodeJS/Typescript

• Docker