Staff Machine Learning Engineer

The role

You will research, build and deploy reinforcement learning methods for planning in autonomous vehicles. You will work with computer vision, AI systems and software engineers, and use data from an autonomous fleet.

What you will do

• Research and develop reinforcement learning based planning algorithms.
• Design model architectures that run in real time on vehicle computing platforms.
• Develop offline data-driven ML infrastructure for fast adaptation of planning models.
• Deliver planning software and work closely with the perception team.
• Work with large field-testing data to improve autonomous driving technologies.
• Design, run and analyze experiments on real-world data.
• Partner with system software engineers to ship industrial-strength ML models.
• Communicate and collaborate with multi-functional teams.

Required qualifications

• Education in Engineering, Robotics or Computer Science with a focus on Reinforcement Learning, Artificial Intelligence or a related field, or equivalent experience.
• Strong experience in applied reinforcement learning, including model architecture design, model training, data mining and data analytics.
• 3-5+ years of experience with machine learning frameworks such as PyTorch or TensorFlow.
• Strong Python programming experience with software design skills.
• Solid understanding of data structures, algorithms, code optimization and large-scale data processing.
• Excellent problem-solving skills.

Nice to have

• MS or PhD level education in Engineering, Robotics or Computer Science with a focus on Reinforcement Learning, Artificial Intelligence or a related field, or equivalent experience.
• Related experience in autonomous driving or robotics production solutions.
• Hands-on experience developing RL based planning engines for autonomous driving or robotic systems.
• Experience with model deployment and optimization tools such as ONNX and TensorRT.

Benefits

• A fun, supportive and engaging environment.
• Infrastructure and computational resources to support your work.
• Opportunity to work on cutting-edge technologies with top talent in the field.
• Opportunity to make a significant impact on the transportation revolution by advancing autonomous driving.
• Competitive compensation package.
• Snacks, lunches, dinners and fun activities.

Before you apply

• The posting states it is an Equal Opportunity Employer.

About the company

XPeng Motors is one of China's leading smart electric vehicle companies. It designs, develops and manufactures smart EVs integrated with Internet, AI and autonomous driving technologies.