Senior ML Engineer, Motion Planning for Autonomous Trucks

PlusAI, Inc.; Santa Clara (CA); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Catered lunch

Unlimited snacks and beverages

401(k) plan

Job summary

PlusAI, Inc. is seeking a Senior/Staff Machine Learning Engineer to train and deploy ML-based planning algorithms for autonomous trucks. You will work with rich map, perception, routing and sensor data to build trajectory generation and behavior planning models.

You will own data curation, experimentation, training, deployment, monitoring and continuous improvement in a production environment. You will collaborate with runtime, perception, mapping and systems teams to ensure safety, robustness

Qualifications

• BS, MS, or PhD in Computer Science, Robotics, ML or related field.
• 4+ years building ML systems for robotics or real-time decision making.
• Strong Python and PyTorch experience.
• Expertise in deep learning, sequence models, transformers or diffusion models.
• Experience designing datasets, experiments and metric-driven evals.
• Production-grade software engineering experience.
• Strong debugging and analytical skills for complex data and models.
• Experience validating models and improving robustness via experiments.
• Ability to drive projects end-to-end with cross-functional teams.
• Excellent communication across engineering teams.

Responsibilities

• Develop ML models for autonomous vehicle planning using map, perception, routing and sensor data.
• Design architectures for trajectory generation and behavior planning.
• Own end-to-end ML lifecycle: data curation, feature engineering, training and deployment.
• Create offline eval strategies and metrics for safety and robustness.
• Analyze model behavior and improve performance via targeted experiments.
• Collaborate with runtime, perception, mapping and systems teams to deploy ML in production.
• Design validation strategies to ensure safe, feasible trajectories.
• Stay current with ML and autonomy advances and translate research into production.
• Ensure work complies with QMS, customer requirements and standards.

Skills

Python programming

Deep learning

PyTorch

ML systems

Transformers

Experimentation

Production software

Cross-functional work

Debugging

System optimization

Education

CS/Robotics degree

Tools

TensorRT

ONNX Runtime

CUDA

C++ (production)

Cloud ML infrastructure

Job description

PlusAI, Inc. is seeking a Senior/Staff Machine Learning Engineer to train and deploy ML-based planning algorithms for autonomous trucks. You will work with rich map, perception, routing and sensor data to build trajectory generation and behavior planning models.

You will own data curation, experimentation, training, deployment, monitoring and continuous improvement in a production environment. You will collaborate with runtime, perception, mapping and systems teams to ensure safety, robustness