2 days ago
Mountain View, CA, USAMid Level
H1B sponsor
Base Salary
$139k - $223k/yr
Responsibilities
- Define the architecture of onboard planning models for autonomous driving.
- Develop and deploy large-scale models trained with Imitation Learning and Reinforcement Learning.
- Build offboard critic models to evaluate comfort, progress, and safety in driving behavior.
- Architect foundation models and world models for the Aurora Driver’s simulation engine.
- Advance autonomous-driving research while deploying models on production vehicles operating on public roads.
- Train large models on massive datasets using distributed computing and write high-performance, maintainable code.
Requirements
- Bachelor’s, master’s, or doctoral degree in Robotics, Machine Learning, Computer Science, or a related quantitative field, or equivalent practical experience.
- At least 2 years of experience developing state-of-the-art ML models in research or production settings.
- Hands-on experience with Imitation Learning or Reinforcement Learning applied to physical or simulated agents.
- Experience training large models on massive datasets using distributed computing.
- Fluency in Python and deep experience with PyTorch or another modern ML framework.
- Mastery of modern ML architectures including Transformers and Diffusion Models.
- Preferred qualifications include top-tier ML conference publications, production ML deployment experience, generative-model or neural-simulator development, and C++ fluency.
Benefits
- Annual bonus, equity compensation, and benefits are available in addition to base salary.
- Aurorans are expected to work from the office at least 3 days per week, with some roles requiring up to 5 days per week based on operational needs.
Categories
ML EngineeringRobotics
About Aurora
Aurora builds the Aurora Driver, a self-driving system for heavy-duty trucks, and offers it as a driver-as-a-service to freight carriers. The company partners with truck OEMs and logistics networks, including Volvo Trucks, PACCAR, FedEx, Uber Freight, Werner, Schneider, and Ryder, to commercialize autonomous freight hauling. Founded in 2017 and headquartered in Pittsburgh, Aurora is a public company trading on Nasdaq under the symbol AUR.