Lead ML Engineer, Robotics & Autonomy - Remote (Canada)

Aimlroles; Los Angeles, Northern (CA, KY); Hybrid; USD 180,000 - 260,000; Full time

Job summary

Serve Robotics is seeking a senior ML engineer to design and scale large-scale training systems for multimodal robotics data, enabling high-performance autonomy models. You will optimize distributed pipelines, neural architectures, and data processing to improve training efficiency and GPU utilization.

The role collaborates with ML researchers and infrastructure teams to deploy models in production and guide model evolution through experiments and metrics.

Qualifications

• Master’s or PhD in Computer Science, Robotics, Electrical Engineering, Machine Learning, or related field.
• 5+ years of professional experience developing, training, and deploying ML models in production.
• Hands-on experience training ML models across multiple GPUs with distributed training frameworks and large datasets.
• Strong programming skills in Python for ML pipelines and training workflows.
• Solid knowledge of neural networks, optimization algorithms, loss functions, model evaluation, and training methodologies.

Responsibilities

• Design and maintain training systems that can process and learn from petabyte-scale multimodal datasets (e.g., video and point cloud data).
• Identify and resolve bottlenecks in the training pipeline to maximize GPU utilization and reduce training time.
• Work with the ML team to develop and refine neural network architectures for autonomy tasks.
• Create and adjust loss functions and training strategies to improve autonomy performance.
• Configure, monitor, and maintain large-scale distributed training jobs across multiple machines and GPUs.
• Implement scalable systems to preprocess, transform, and augment large robotics datasets for model training.
• Collaborate with ML scientists and engineers to integrate new models into production training pipelines.
• Analyze training metrics and experiment logs to guide improvements in architecture and data usage.
• Develop tools and workflows to run experiments, track results, and iterate on new model ideas.

Skills

Python programming

Multi-GPU training

Distributed training

Neural networks

Optimization algorithms

Training methodologies

Education

Master’s or PhD in Computer Science, Robotics, Electrical Engineering, Machine Learning, or related

Job description

Serve Robotics is seeking a senior ML engineer to design and scale large-scale training systems for multimodal robotics data, enabling high-performance autonomy models. You will optimize distributed pipelines, neural architectures, and data processing to improve training efficiency and GPU utilization.

The role collaborates with ML researchers and infrastructure teams to deploy models in production and guide model evolution through experiments and metrics.