Sr. Software Engineer, ML Edge Inference

Sr. Software Engineer, ML Edge Inference Engineer

Join to apply for the Sr. Software Engineer, ML Edge Inference Engineer role at Serve Robotics.

$/yr - $/yr

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.

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.

We are seeking a highly skilled Sr. Software Engineer, ML Edge Inference Engineer to join our robotics team. This technical role bridges the gap between ML research and real-time deployment, enabling advanced ML models to run efficiently on edge hardware such as NVIDIA Jetson platforms. You will work closely with ML researchers, embedded systems engineers, and robotics software teams to ensure that state-of-the‑art models can be deployed with optimal performance on robotic platforms.

• Own the full lifecycle of ML model deployment on robots—from handoff by the ML team to full system integration.
• Convert, optimize, and integrate trained models (e.g., PyTorch/ONNX/Tensor
  
  RT) for Jetson platforms using NVIDIA tools.
• Develop and optimize CUDA kernels and pipelines for low-latency, high-throughput model inference.
• Profile and benchmark existing ML workloads using tools like Nsight, nvprof, and Tensor
  
  RT profiler.
• Identify and remove compute and memory bottlenecks for real‑time inference.
• Design and implement strategies for quantization, pruning, and other model compression techniques suited for edge inference.
• Ensure models are robust to the resource constraints of real‑time, low‑power robotic systems.
• Manage memory layout, concurrency, and scheduling for optimized GPU and CPU usage on Jetson devices.
• Build benchmarking pipelines for continuous performance evaluation on hardware‑in‑the‑loop systems.
• Collaborate with QA and systems teams to validate model behavior in field scenarios.
• Work closely with ML researchers to influence model architectures for edge deployability and provide technical guidance on the feasibility of real‑time ML models in the robotics stack.

• Bachelor’s degree in Computer Science, Robotics, Electrical Engineering, or equivalent field.
• 5+ years experience in deploying ML models on embedded or edge platforms (preferably robotics).
• 3+ years of experience with CUDA, Tensor
  
  RT, and other NVIDIA acceleration tools.
• Proficient in Python and C++, especially for performance‑sensitive systems.
• Experience with NVIDIA Jetson (e.g., Xavier, Orin) and edge inference tools.
• Familiarity with model conversion workflows (e.g., PyTorch → ONNX → Tensor
  
  RT).

Please note:

The base salary range listed in this job description reflects compensation for candidates based in the San Francisco Bay Area. While we prefer candidates located in the Bay Area, we are also open to qualified talent working remotely across the United States. Base salary range (U.S. – all locations): $180,000 – $205,000.

• Master’s degree in Computer Science, Robotics, Electrical Engineering, or equivalent field.
• Experience with real‑time robotics systems (e.g., ROS2, middleware, safety‑critical constraints and Linux embedded systems).
• Knowledge of performance tuning under thermal, power, and memory constraints on embedded devices.
• Experience with model quantization (e.g., INT8), sparsity, and latency‑aware model design.
• Contributions to open‑source ML or CUDA projects is a plus.

Mid‑Senior level

Full‑time

Engineering and Information Technology

Technology, Information and Internet

Referrals increase your chances of interviewing at Serve Robotics by 2x.