Serve Robotics Logo

Serve Robotics

Lead Machine Learning Engineer

Reposted 3 Months Ago
In-Office or Remote
2 Locations
Senior level
In-Office or Remote
2 Locations
Senior level
Develop and scale machine learning training systems for robotics data, optimize training pipelines, and collaborate with ML researchers to enhance autonomy models.
The summary above was generated by AI

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.

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.

  • 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.

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 such as camera video, LiDAR point clouds, radar, or telemetry data.

  • Experience developing models that combine multiple data modalities (e.g., images, point clouds, and structured sensor data) into a unified learning system.

  • Peer-reviewed publications or significant research contributions in machine learning, robotics, or related areas.

*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

Similar Jobs

Yesterday
Easy Apply
Remote or Hybrid
Easy Apply
Expert/Leader
Expert/Leader
Artificial Intelligence • Big Data • Computer Vision • Information Technology • Machine Learning • Analytics • Defense
Leads the design, development, and delivery of production machine learning and AI systems for government and commercial customers. Responsibilities include data shaping, analytics, model development and optimization, production deployment, technical planning, customer engagement, and mentoring cross-functional engineering teams. The role requires translating complex technical concepts for internal and external stakeholders and traveling up to 25% of the time.
Top Skills: Cloud EnvironmentsMachine LearningPython
6 Days Ago
In-Office or Remote
Senior level
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Leads the design, development, and deployment of AI and machine learning solutions for healthcare technology platforms. Builds scalable ML models, MLOps pipelines, LLM and RAG systems, backend services, and agent-based applications. Establishes standards for monitoring, testing, CI/CD, and lifecycle management while collaborating with engineering, data, product, security, and SRE teams. Provides architectural guidance and technical mentorship, evaluates emerging AI technologies, and promotes responsible, secure, and resilient AI implementation.
Top Skills: AWSAzureAzure Ai SearchCi/CdCrewaiDockerEmbeddingsFaissGoogle Cloud PlatformJavaJavaScriptLangchainLanggraphLarge Language ModelsLlamaindexMcpMilvusMlopsModel Fine-TuningPeftPgvectorPineconePrompt EngineeringPythonRetrieval-Augmented GenerationSemantic Kernel
6 Days Ago
In-Office or Remote
Senior level
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Leads the design, development, deployment, and operationalization of enterprise AI solutions, including LLM applications, RAG systems, intelligent agents, classification models, and decision-support tools. Establishes prompt strategies, guardrails, evaluation frameworks, monitoring, observability, and governance practices. Partners with Finance, Data Engineering, Architecture, Security, and Responsible AI teams to deliver scalable, secure, compliant, and production-ready solutions.
Top Skills: Ai AgentsAi ObservabilityAPIsArtificial IntelligenceCloud-Native ArchitecturesData PlatformsLarge Language Models (Llms)Machine LearningModel EvaluationModel MonitoringPrompt EngineeringRetrieval-Augmented Generation (Rag)Semantic SearchVector Databases

What you need to know about the Calgary Tech Scene

Employees can spend up to one-third of their life at work, so choosing the right company is crucial, not just for the job itself but for the company culture as well. While startups often offer dynamic culture and growth opportunities, large corporations provide benefits like career development and networking, especially appealing to recent graduates. Fortunately, Calgary stands out as a hub for both, recognized as one of Startup Genome's Top 100 Emerging Ecosystems, while also playing host to a number of multinational enterprises. In Calgary, job seekers can find a wide range of opportunities.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account