Lead Machine Learning Engineer

Serve Robotics, USA (remote) +5

What you will do

Design and maintain training systems that can process and learn from petabyte-scale multimodal datasets (e.g., video and point cloud data). This includes ensuring data is efficiently loaded, distributed, and processed across large GPU clusters.
Identify and resolve bottlenecks in the training pipeline, including data loading, preprocessing, model computation, and inter-node communication, to maximize GPU utilization and reduce training time.
Work with the ML team to develop and refine neural network architectures suitable for autonomy tasks, particularly those handling high-dimensional and sequential sensor data.
Create and adjust loss functions and training strategies that help the model learn effectively from complex multimodal inputs and improve autonomy performance.

Posted pay

$225K–$260K

a year

This base pay is above the median for senior ML and AI engineering jobs that are remote in the US.

We compare the middle of each posted range. This is base pay as posted. The group does not include this job.

Location

USA (remote) +5

Work type

Remote

Level

Senior

Experience

5+ years

Skills in the job text

Posted

Mar 27, 2026

Last check

8 h ago

Job type

ML / AI engineering, Inference & ML systems

What you will do

Design and maintain training systems that can process and learn from petabyte-scale multimodal datasets (e.g., video and point cloud data). This includes ensuring data is efficiently loaded, distributed, and processed across large GPU clusters.
Identify and resolve bottlenecks in the training pipeline, including data loading, preprocessing, model computation, and inter-node communication, to maximize GPU utilization and reduce training time.
Work with the ML team to develop and refine neural network architectures suitable for autonomy tasks, particularly those handling high-dimensional and sequential sensor data.
Create and adjust loss functions and training strategies that help the model learn effectively from complex multimodal inputs and improve autonomy performance.