ashby
Lead Machine Learning Engineer
Serve Robotics
Remote, CAN
Full-time
Senior · 5+ years experience
Remote
USD 225000-260000 yearly / year
Discovered 3 weeks ago
Python
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
Python
Key Skills for This Role
Python
Full Job Posting
Who We Are
We are tech industry veterans in software, hardware, and design who are pooling our skills to build the future we want to live in. We are solving real-world problems leveraging robotics, machine learning and computer vision, among other disciplines, with a mindful eye towards the end-to-end user experience. Our team is agile, diverse, and driven. We believe that the best way to solve complicated dynamic problems is collaboratively and respectfully.
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). 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.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Serve Robotics
Senior Corporate Strategist
, USA
SeniorFull-time
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
Discovered TodayView →
Director, Strategic Partnerships and New Ventures
, USA
DirectorFull-time
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
Discovered TodayView →
Operations Associate (North Hollywood, Los Angeles)
Los Angeles, USA
EntryFull-time
Discovered 1 weeks agoView →
Operations Associate (Downtown, Los Angeles)
Los Angeles, USA
EntryFull-time
Discovered 1 weeks agoView →
Senior Corporate Strategist
, USA
Discovered TodayFull-time
Director, Strategic Partnerships and New Ventures
, USA
Discovered TodayFull-time
VP Product & Design
, USA
Discovered 3 days agoFull-time
VP of IT Infrastructure & Operations
, CAN
Discovered 3 days agoFull-time
Operations Associate (North Hollywood, Los Angeles)
Los Angeles, USA
Discovered 1 weeks agoFull-time
Operations Associate (Downtown, Los Angeles)
Los Angeles, USA
Discovered 1 weeks agoFull-time
Operations Coordinator
Miami, USA
Discovered 1 weeks agoFull-time
Operations Associate - Part Time
Los Angeles, USA
Discovered 1 weeks agoPart-time