Lead Machine Learning Engineer

Lead Machine Learning Engineer in Canada and USA

Serve Robotics is seeking a senior-level Lead Machine Learning Engineer for an IT role focused on developing and scaling large-scale machine learning training systems for multimodal robotics data. In this position, you will optimize distributed training pipelines, neural network architectures, and data processing workflows to improve training efficiency, accelerate model iteration, and maximize GPU utilization across the fleet.

Core Responsibilities

• Design and maintain training systems that process petabyte-scale multimodal datasets like video and point cloud data.
• Identify and resolve bottlenecks in training pipelines to maximize GPU utilization and reduce training time.
• Collaborate with the machine learning team to refine neural network architectures for autonomy tasks handling high-dimensional sensor data.
• Configure and monitor large-scale distributed training jobs across multiple machines and GPUs to ensure stability and fault tolerance.
• Implement scalable systems to preprocess, transform, and augment large robotics datasets for model training.

Requirements and Qualifications

Candidates must hold a PhD or specialized diploma, specifically a Master’s or PhD in Computer Science, Robotics, Electrical Engineering, Machine Learning, or a closely related technical discipline. Applicants are required to have a minimum of 5 years of professional experience developing, training, and deploying machine learning models in production environments. Strong programming skills in Python, hands-on experience with distributed training across multiple GPUs, and solid knowledge of neural networks and optimization algorithms are essential for this role. The listed salary is $177, and specific details regarding application contact methods or exact regional breakdowns beyond Canada and the USA are not provided in the source text.