Lead Machine Learning Engineer
Remote$225k - $260kFull-timePosted 3mo ago
About the role
Role focused on building and scaling large-scale distributed machine learning training systems for multimodal robotics data to enable high-performance autonomy models, collaborating with ML researchers and infrastructure teams.
Responsibilities
- 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.
- Configure, monitor, and maintain large-scale distributed training jobs across multiple machines and GPUs, ensuring stability, fault tolerance, and efficient resource usage.
- Implement scalable systems to preprocess, transform, and augment large robotics datasets so that they are suitable for model training.
- 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 in architecture, data usage, or training strategies.
- 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, including familiarity with distributed training frameworks and large dataset handling.
- 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.
Skills
Required:Multimodal data handlingDistributed trainingPython
Last updated Jul 16, 2026. Interested? Apply on the company site.
About Serve Robotics
The future is here
Why move a 2-pound burrito in a 2-ton car? Meet Serve, the future of sustainable, self-driving delivery.
About UsAdvertisingAutonomous deliveryCareersDataDelivery SolutionsFAQsInvestors
More open roles at Serve Robotics
- Service Technician II3d agoMiami, FL · Operations
- Manager, Strategic Finance3d agoRemote · Finance
- Operations Associate (Midtown, Atlanta)1w agoAtlanta, GA · Operations
- Financial Analyst, FP&A1w agoRemote · Finance
- Service Technician II1w agoAtlanta, GA · Operations