ML Infrastructure Engineer

Sunday Robotics

Redwood City (CA)

On-site

USD 120,000 - 150,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

A robotics innovation company focused on home robotics seeks a Software Engineer to develop machine learning infrastructure. You will own training systems, optimize data pipelines, and work on real-time robot control systems. Ideal candidates have a strong software engineering background, with experience in distributed systems, machine learning infrastructure, and collaboration with researchers. This role offers the chance to contribute to groundbreaking robotics technology in an inclusive and diverse environment.

Qualifications

  • Strong software engineering principles and problem-solving skills.
  • Experience in creating and managing complex data pipelines.
  • Familiarity with machine learning frameworks, especially PyTorch.

Responsibilities

  • Maintain an effective research codebase optimizing for fast iteration and correctness.
  • Own infrastructure for training: job scheduling and metrics.
  • Scale distributed training across GPU clusters with minimal friction.
  • Build low-latency inference pipeline for real-time control of robots.
  • Design high-throughput data pipelines for multimodal robot data.

Skills

Strong software engineering and systems fundamentals
Experience building distributed systems or large-scale data pipelines
Hands-on experience with ML training infrastructure, ideally PyTorch
Comfort reasoning about performance, memory, I/O, and GPU utilization
Experience managing training workloads (SLURM, Kubernetes, or similar)

Job description

Join Us in Building the Future of Home Robotics

At Sunday, we're developing personal robots to reclaim the hours lost to repetitive tasks. We're focused on an ambitious goal to make generalized robots broadly accessible, enabling households to take back quality time.

We have spent the last 18 months building a talented team, securing capital, and validating our technology. We are now seeking passionate individuals to join us in the next phase of our growth. If you are ready to apply your skills to the forefront of robotics innovation, we’d love to hear from you.

The Role

Sunday Robotics is building the future of home robotics. We're developing end-to-end ML models for robot manipulation, and you'll have the opportunity to build and shape foundational systems that directly accelerate our path to putting robots in homes.

This is a broad role that can be tailored to your specific area of expertise: data pipelines, training infrastructure or inference. You'll build systems across the full robot learning pipeline: ingesting and processing multimodal data, scaling distributed training, optimizing inference for real-time control and building research tooling.

What You'll Do
Training and Inference Infrastructure
  • Maintain an effective research codebase with good ergonomics, optimizing for fast iteration and correctness
  • Own infrastructure for model training: job scheduling, checkpointing, metrics, and logging
  • Scale distributed training across GPU clusters with minimal researcher friction
  • Enable training of larger models through sharding, activation checkpointing and memory optimization
  • Profile and optimize GPU utilization, memory usage and training throughput
  • Build low-latency inference pipeline for real-time robot control, apply quantization, distillation and model compilation to optimize inference performance
  • Work closely with researchers and roboticists to translate research needs into reliable software and infrastructure
Data Pipelines and Research Tooling
  • Design high-throughput pipelines for ingesting, validating, and transforming multimodal robot data (video, proprioception, actions)
  • Build storage systems and metadata indexing for efficient dataset management at large scale
  • Optimize dataloaders, sharding and prefetching to minimize time from data arrival to model training
  • Build research tooling for debugging, visualization and experiment analysis
What We're Looking For
  • Strong software engineering and systems fundamentals
  • Experience building distributed systems or large-scale data pipelines
  • Hands‑on experience with ML training infrastructure, ideally PyTorch
  • Comfort reasoning about performance, memory, I/O, and GPU utilization
  • Experience managing training workloads (SLURM, Kubernetes, or similar)
  • Ownership mindset: you design, build, operate, and iterate on systems end-to-end
  • Enjoy working closely with researchers and unblocking fast-moving projects
Nice to Have
  • Experience with robotics data pipelines or multimodal models
  • Background in VLAs, Video Generation architectures or robot learning systems
  • Deep ML systems experience: training compilers, custom kernels, runtime optimization
  • Hands‑on GPU performance tuning
  • Experience with serialization formats for high-performance systems (Protobuf, FlatBuffers, MCAP)

At Sunday Robotics, we’re building technology shaped by real people — curious, creative, and diverse. We’re proud to be an equal opportunity employer and consider all qualified applicants regardless of race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, or veteran status.

Even if you don’t meet every single requirement, we encourage you to apply. Studies show that women and underrepresented groups often hold back unless they meet 100% of the criteria — we don’t want that to be the reason we miss out on great talent.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

ML Infrastructure Engineer
ML Infrastructure Engineer

Sunday • Redwood City (CA)

On-site
USD 100,000 - 130,000
Systems Software Engineer, ML & Robotics Infra
Systems Software Engineer, ML & Robotics Infra

Dormont Manufacturing Co • Redwood City (CA)

On-site
USD 110,000 - 150,000
Developer Infrastructure Engineer
Developer Infrastructure Engineer

Sunday • Redwood City (CA)

On-site
USD 120,000 - 160,000
Machine Learning Research Engineer/Scientist
Machine Learning Research Engineer/Scientist

Sunday Robotics • Mountain View (CA)

On-site
USD 150,000 - 210,000
System Software Engineer — GPU & Accelerated Compute
System Software Engineer — GPU & Accelerated Compute

Sunday • Redwood City (CA)

On-site
USD 100,000 - 130,000
Systems Software Engineer
Systems Software Engineer

Sunday • Redwood City (CA)

On-site
USD 100,000 - 130,000
Robotics Systems Engineer
Robotics Systems Engineer

Sunday • Redwood City (CA)

On-site
USD 80,000 - 120,000
Embedded Software Engineer , ML & Robotics Infra Redwood City, CA
Embedded Software Engineer , ML & Robotics Infra Redwood City, CA

Sunday Inc. • Redwood City (CA), Northern (KY)

Hybrid
USD 120,000 - 150,000
Technical Program Manager, Data Engine
Technical Program Manager, Data Engine

Sunday • Redwood City (CA)

On-site
USD 90,000 - 120,000
Robotics Controls Engineer , Robotics Mountain View
Robotics Controls Engineer , Robotics Mountain View

Sunday Inc. • Redwood City (CA)

On-site
USD 90,000 - 130,000