Skip to content
Bots n' Beans
Sign in

Jobs

Nimble Robotics

Robot type
Warehouse and Logistics
Location
San FranciscoCaliforniaUSA
Job type
Artificial Intelligence
Posted
Aug 18, 2026
Full-time

Senior/Staff Software Engineer, Infrastructure (ML)

Job description

We are on a mission to empower and inspire mankind to accomplish legendary feats by inventing robots that liberate us from the menial. We will accomplish this by training robot AGI to invent and build the Autonomous Supply Chain – everything from the inside of factories and warehouses to your front door – powered by generalist superhumanoids.

Our founding team comes from the AI labs at Stanford and Carnegie Mellon and our board of directors include famed robotics and AI legends including Fei-Fei Li (Chief Scientist of AI at Google and Director of Stanford’s AI Lab), Marc Raibert (founder of Boston Dynamics), and Sebastian Thrun (founder of GoogleX, Waymo;

Job responsibilities

  • Design, develop, and maintain ML training infrastructure that enables the AI team to run training jobs efficiently, manage and iterate experiments quickly.
  • Build low-latency inference pipelines for production robotics workloads.
  • Develop, tune, and optimize low-level CUDA kernels.
  • Design training-platform systems for scalable model training, including high-throughput data ingestion, dataset sharding and sampling for distributed training.
  • Participate in and lead design reviews with peers and stakeholders to evaluate technical tradeoffs and select appropriate technologies.
  • Review code and provide feedback to uphold best practices around style, correctness, testability, performance, and maintainability.
  • Contribute to documentation and educational materials, adapting content as systems and workflows evolve.
  • Mentor junior engineers and help raise the technical bar across the team.

Job requirements

  • Bachelor’s, Master’s, or PhD in Computer Science or a related field, or equivalent practical experience.
  • 4+ years of industry experience in infrastructure, distributed systems, ML systems, robotics, or a related area.
  • Experience with programming languages such as Rust, Go, Python, or C++.
  • Experience with ML frameworks such as PyTorch or JAX.
  • Strong understanding of distributed systems, systems programming fundamentals, memory management, and performance optimization.
  • Experience with Kubernetes orchestration, resource scheduling for large distributed jobs, and containerized deployment pipelines.
  • Ability to debug and optimize bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operations.
  • Ability to reason from first principles and optimize systems for both memory-bound and compute-bound workloads.
Apply on Nimble Robotics websiteSave

Similar jobs