Senior Staff ML Engineer, Driver Understanding and Evaluation

What you'll do

• Develop strategy and evaluation metrics for machine-learning-based evaluation, emphasizing scientific and statistical rigor.
• Architect scalable training systems and lead development of generative models, RL algorithms, and evaluation workflows for driving behaviors.
• Provide technical leadership and mentorship, aligning collaborating teams on evaluation goals and priorities.

What they ask for

• PhD in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
• 10+ years of hands-on experience developing and applying machine-learning models, with significant focus on Reinforcement Learning.
• 2+ years of people management experience.
• Expertise in deep learning, sequence modeling, and generative models; proficiency in Python and standard ML frameworks (e.g., JAX, TensorFlow).

Nice to have

• Experience in autonomous vehicles, robotics, or complex simulation environments.
• Experience with large-scale simulation platforms, evaluation metrics for complex AI systems, or RL fine-tuning using human feedback/preferences.

Pay

$281,000 — $356,000 USD base salary

Tools

Python, JAX, TensorFlow, LLM/VLM, Reinforcement Learning from Human Preference (RLHF)

AI focus

Builds generative AI and deep-learning solutions, RL algorithms, and RLHF-based data collection and evaluation systems to generate, analyze, and assess autonomous driving behaviors.