Machine Learning Engineer (Simulation Realism)

TL;DR

Machine Learning Engineer (Simulation Realism): Developing ultra-realistic world simulations for autonomous vehicles using foundation models for embodied agents with an accent on generative AI and world modeling. Focus on transferring AI research into scalable production solutions, enhancing simulation fidelity, and testing autonomous driving systems in counterfactual scenarios.

Location: Hybrid / Onsite in Mountain View, California

Salary: $204,000—$259,000 USD

Company

Autonomous driving technology company building the hirify.global Driver for fully autonomous ride-hail services and other vehicle platforms.

What you will do

• Drive innovation in simulation technologies using foundation models for realistic embodied agents like vehicles, pedestrians, and cyclists.
• Apply ML expertise in Generative AI to enhance simulation realism for roads, traffic, and weather.
• Collaborate with AI research team to productionize world understanding, reasoning, and reinforcement learning solutions.
• Build infrastructure for counterfactual scenarios to test autonomous driving performance.

Requirements

• 5+ years of experience with ML software engineering
• 5+ years of experience with Python
• 5+ years of ML software experience with frameworks including Pytorch, Tensorflow, Jax/Flax

Nice to have

• Experience with ML models for robotics or self-driving vehicles
• Strong C++ in production, integrating ML models
• Training large-scale models on GPU/TPU clusters

Culture & Benefits

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