Machine Learning Software Engineer, Motion Planning: Behavior

$144,000 - $216,000/Yr

Aurora Innovation - San Francisco, CA

Full-time - Mid Level

San Francisco, 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 to manage interactions among multiple actors while accounting for uncertainties in the environment. You will also work on integrating motion planning with the forecasting of other agents, ensuring that our self-driving systems can navigate complex scenarios effectively. In this position, you will be tasked with developing and implementing production-grade algorithms that will be integral to the Driver system. Your work will directly contribute to the safety and efficiency of self-driving technology, which is designed to operate across various vehicle types, from freight-hauling semi-trucks to ride-hailing passenger vehicles. You will collaborate with a diverse team of talented individuals who are committed to building a transportation ecosystem that enhances road safety and accessibility for all. This role requires a strong foundation in robotics and machine learning, as well as practical experience in software development within a production environment. You will leverage your programming skills, particularly in C++ and Python, to create innovative solutions that push the boundaries of what is possible in autonomous vehicle 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 system.

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 eligibility
• Equity compensation
• Comprehensive benefits package