Senior/Staff Software Engineer, Infrastructure (ML)
Core
Building end-to-end training and inference systems for general-purpose warehouse robots using GPU clusters.
Role type
Senior/Staff IC ML Infrastructure Engineer
Builds
Scalable ML training platforms and low-latency inference pipelines for robotics workloads
Domain
Robotics + Distributed Systems
Deliverable
infrastructure
Required skills
Rust, Go, Python, C++, PyTorch, JAX, CUDA kernel development, Kubernetes, distributed systems, memory management, performance optimization
Preferred skills
PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL, GPU kernel development, data engineering (Parquet, Arrow)
Technologies
PyTorch, JAX, CUDA, Kubernetes, Rust, Go, Python, C++
Sourced via greenhouse · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.