Watney
Staff Software Engineer - ML Infrastructure
San Francisco · On-site · Staff · Full-time · Posted Jul 13, 2026
About the job
Build and scale ML infrastructure that turns data from a live robot fleet into trainable data and supports model training and inference.
What you'll do
- Own training and inference infrastructure.
- Build data pipelines and make experiments fast to launch and reproduce.
- Contribute to core training code.
What they ask for
- Experience building ML infrastructure used for production training runs.
- Experience scaling distributed training systems.
- Strong experience with Python and PyTorch or TensorFlow.
- Experience identifying and troubleshooting GPU performance bottlenecks in large-scale training environments.
Tools
Python, PyTorch, TensorFlow, GPU
AI focus
Support machine-learning model training and inference using video and telemetry data collected by a live robot fleet; scale training runs and improve GPU utilization.
Model-written summary · the original posting is authoritative