Machine Learning Engineer for Motion and Behavior Planning

About the role

Tensor is seeking a skilled Machine Learning Engineer to join our dynamic team. The ideal candidate will have a strong background in computer science, robotics, or a related field, with expertise in Python and C++, as well as experience with PyTorch, TensorFlow, and large-scale ML pipelines.

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.
• Create tools to benchmark and analyze GPU performance.

Qualifications

• An MS in Computer Science, Robotics, or a similar technical field, or equivalent practical experience.
• Strong programming skills in Python and/or C++.
• Experience with PyTorch, TensorFlow, and their computation graphs.
• Experience in building large-scale ML pipelines.
• Good communication and collaboration skills.

Preferences

• A PhD in Computer Science, Robotics, or a similar technical field, or equivalent practical experience.
• Experience with developing ML-based algorithms for prediction, motion planning, or Large Language Models (LLMs)/Vision-Language Models (VLMs).
• Experience with ML model compression, quantization, and distillation.
• Experience with GPU profiling, GPU scheduling.
• At least 2 years of industrial experience.

Benefits

We offer a competitive compensation package, opportunities for professional growth, participation in a discretionary equity incentive plan, and access to a comprehensive company benefits program, subject to eligibility requirements.

Pay

$75k—$300k USD

Schedule

Full-time

Locations

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