Machine Learning Engineer for Motion and Behavior Planning

AutoX is a global leader in self-driving car technology, driven by our mission to "Democratize Autonomy" and accelerate the advent of fully driverless cars. Our cutting-edge self-driving cars are designed to tackle the most complex and dynamic traffic environments, from bustling city streets to ever-changing urban landscapes. In 2020, we became the second company to secure a fully driverless passenger vehicle permit in California — a key milestone in autonomy. With a world-class team of engineers and R&D centers across the U.S. and beyond, AutoX continues to innovate.

Join us to shape the future! Work with brilliant minds on technology transforming transportation and make a lasting impact.

Locations:

• San Jose, California, US (Salary Range: $75k—$300k USD)
• Barcelona, Catalonia, Spain
• Singapore
• Dubai, UAE

Responsibilities:

• Develop ML models and algorithms for motion and behavior planning for autonomous vehicles.
• Build ML pipelines that automate the entire ML lifecycle, including data extraction, model training, and evaluation.
• Coordinate the training and release of multiple dependent models.
• Build tools to benchmark and analyze GPU performance.

Qualifications:

• MS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience.
• Strong in Python and/or C++ programming skills.
• Experience in PyTorch/Tensorflow and its computation graph.
• Experience with building large-scale ML pipelines.
• Good communication and collaboration skills.

Preferences:

• PhD in Computer Science, similar technical field of study, or equivalent practical experience.
• Experience with developing ML based algorithms for prediction, motion planning or LLM/VLMs.
• Experience with ML model compression, quantization, distillation.
• Experience with GPU profiling, GPU scheduling.
• 2+ years of industrial experience.

How to Apply:

• Send an email with your resume to careers@autox.ai with Job ID: 05

AutoX is an equal employment opportunity employer, committed to maintaining a supportive and inclusive workplace for all employees. AutoX does not discriminate against, and prohibits harassment of, any applicant or employee based on race, color, sex, sexual orientation, gender identity, religion, national origin, age, disability, military status, genetic information or any other basis protected by applicable law in the respected countries.