Senior ML Engineer — Autonomous Driving Planning

Plusai; Santa Clara (CA); On-site; USD 140,000 - 190,000; Full time

Benefits offered by this job

Catered lunch & snacks

401(k) plan

Competitive compensation

Job summary

Plus is seeking a machine learning engineer to train and deploy ML-based planning algorithms for its autonomous trucking fleet. You will develop deep learning models from perception data, design safety checks, and ensure real-time performance while collaborating across teams.

Applicants should have 4+ years in ML for robotics, Python and PyTorch experience, and the ability to tackle end-to-end software development.

Qualifications

• BS, MS, or PhD in Software, Robotics, or related field
• 4+ years of machine learning engineering experience for robotics applications
• Experience developing high-quality software from design and implementation to testing and deployment
• Expertise with Python, willingness to do some C++ development as needed
• Experience training with modern frameworks (e.g. PyTorch)
• A rigorous approach to model development: running well-designed experiments, defining suitable training and validation datasets, and evaluating on the right metrics
• Hands-on familiarity with cloud data ingestion pipelines
• Strong communication and collaborative skills

Responsibilities

• Use complex map and perception sensor output to develop novel deep learning models for our planning stack.
• Design critical safety checks to ensure driving trajectories are feasible and follow rules of road.
• Ensure all model development keeps a real-time focus and operates efficiently in compute-constrained environments.
• Track and incorporate the latest multidisciplinary research advancements in a fast-moving field.
• Ensure that your work is performed in accordance with the company’s Quality Management System (QMS) requirements and contribute to continuous improvement efforts.
• Ensure that technical work meets customer requirements, regulatory standards, and company quality policies.

Skills

Machine learning for robotics

Python

C++ development

PyTorch

Experiment design & evaluation

Cloud data ingestion

Communication skills

Education

BS/MS/PhD in Software, Robotics, or related field

Job description

Plus is seeking a machine learning engineer to train and deploy ML-based planning algorithms for its autonomous trucking fleet. You will develop deep learning models from perception data, design safety checks, and ensure real-time performance while collaborating across teams.

Applicants should have 4+ years in ML for robotics, Python and PyTorch experience, and the ability to tackle end-to-end software development.