[Hiring] Lead Machine Learning Engineer @Serve Robotics

Role Description

At Serve Robotics, we’re reimagining how things move in cities. Our personable sidewalk robot is our vision for the future. It’s designed to take deliveries away from congested streets, make deliveries available to more people, and benefit local businesses.

The Serve fleet has been delighting merchants, customers, and pedestrians along the way in Los Angeles, Miami, Dallas, Atlanta, and Chicago while doing commercial deliveries. We’re looking for talented individuals who will grow robotic deliveries from surprising novelty to efficient ubiquity.

This role develops and scales large-scale machine learning training systems for multimodal robotics data, enabling the creation of high-performance autonomy models. By optimizing distributed training pipelines, neural network architectures, and data processing workflows, the position improves training efficiency, accelerates model iteration, and maximizes GPU utilization. The role collaborates closely with ML researchers and infrastructure teams, influencing the design, deployment, and performance of end-to-end autonomy models and the large-scale data pipelines that support them.

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, including data loading, preprocessing, model computation, and inter-node communication.
• Work with the ML team to develop and refine neural network architectures suitable for autonomy tasks.
• Create and adjust loss functions and training strategies that help the model learn effectively from complex multimodal inputs.
• 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.
• Work closely with ML scientists and other engineers to integrate new models, experiments, and training approaches into the production training pipeline.
• Analyze training metrics, model outputs, and experiment logs to assess model performance and guide improvements.
• Develop tools and workflows that allow teams to run experiments, track results, and iterate quickly on new model ideas or training approaches.

Qualifications

• Master’s or PhD in Computer Science, Robotics, Electrical Engineering, Machine Learning, or a closely related technical discipline.
• Minimum of 5 years of professional experience developing, training, and deploying machine learning models in production environments.
• Hands-on experience training machine learning models across multiple GPUs or compute nodes.
• Strong programming skills in Python for implementing machine learning models, data pipelines, and training workflows.
• Solid knowledge of core concepts such as neural networks, optimization algorithms, loss functions, model evaluation, and training methodologies.

What Makes You Stand Out

• Experience identifying and resolving training bottlenecks related to compute utilization, memory usage, and data throughput in machine learning systems.
• Experience training machine learning models on robotics or autonomous driving datasets involving multimodal sensor inputs.
• Experience developing models that combine multiple data modalities into a unified learning system.
• Peer-reviewed publications or significant research contributions in machine learning, robotics, or related areas.

Compensation

Please note: The listed base salary range applies to candidates based in the US. Compensation may vary depending on location, experience, and role alignment. We are open to qualified candidates working remotely in Canada.

• Canada - ALL: $177k - $215k CAD