Machine Learning Software Engineer, Motion Planning: Behavior

$144,000 - $216,000/Yr

Aurora Innovation - Mountain View, CA

Full-time - Mid Level

Mountain View, CA

Professional, Scientific, and Technical Services

About the position

As a Machine Learning Software Engineer on the Motion Planning Team at Aurora, you will play a pivotal role in advancing self-driving technology. Your primary responsibility will be to build machine learning approaches that enhance motion planning capabilities, focusing on creating, generating, and selecting trajectories that ensure both comfort and safety for self-driving vehicles. This involves developing sophisticated policies and plans that effectively manage interactions among multiple actors while accounting for uncertainties in the environment. You will also be tasked with interleaving motion planning with the forecasting of other agents, which is crucial for the safe operation of autonomous vehicles in dynamic settings. In addition to algorithm development, you will implement production-grade solutions that integrate seamlessly into the Driver system, which supports various vehicle types, including freight-hauling semi-trucks and ride-hailing passenger vehicles. Your work will directly contribute to making transportation safer, more reliable, and accessible, aligning with Aurora's mission to revolutionize the transportation ecosystem. You will collaborate with industry leaders and be part of a diverse team dedicated to pushing the boundaries of self-driving technology.

Responsibilities

• Build machine learning approaches for motion planning to create, generate, and select comfortable and safe trajectories for self-driving vehicles.
• Develop policies and plans to manage multi-actor interactions and plans under uncertainty.
• Develop approaches that interleave motion planning with forecasting of other agents.
• Develop and implement production-grade algorithms for the Driver.

Requirements

• BS, MS, or PhD in Robotics, Machine Learning, Applied Mathematics, Computer Science or related field.
• 1+ years of industry experience building software in a production environment.
• Strong programming and software design skills ideally in C++ and/or Python.
• Knowledge of classical and modern ML approaches such as transformer networks, linear models, recurrent models, convolutional neural networks.
• Familiarity with modern ML and GPU toolkits such as PyTorch, TensorFlow, JAX, TensorRT, and/or CUDA.

Nice-to-haves

• Experience with imitation and reinforcement learning and how machine learning can be used in robotics applications.
• Experience in designing, building, and productionizing robotics systems, especially decision making systems, motion planning algorithms, state estimation algorithms, probabilistic modeling, or statistical techniques.

Benefits

• Annual bonus
• Equity compensation
• Comprehensive health benefits