Machine Learning Engineer for Motion and Behavior Planning

AutoX is a leading global self-driving car company. Our mission ‘Democratizing Autonomy’ is to accelerate the advent of fully driverless cars. The company’s self-driving platform is capable of handling the most challenging and dynamic traffic scenarios in urban cities around the world. In 2020, AutoX obtained the world's second fully driverless RoboTaxi permit from the state of California. AutoX has a large team of engineers, with multiple R&D centers in the US and around the world.

Locations:

• San Jose, California, US
• Barcelona, Catalonia, Spain
• Singapore

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

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.