Sr. Software Engineer, ML Edge Inference

• 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/TensorRT) 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 TensorRT profiler.

This role enables advanced ML models to run efficiently on robotic platforms, helping transform robotic deliveries from novelty to efficient ubiquity while taking deliveries away from congested streets and making them available to more people.

• 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, TensorRT, and other NVIDIA acceleration tools.
• Proficient in Python and C++, especially for performance-sensitive systems.

Base salary range (U.S. – all locations): $180,000 – $205,000

Serve Robotics is reimagining how things move in cities through personable sidewalk robots designed to take deliveries away from congested streets, make deliveries available to more people, and benefit local businesses.