← All roles

Staff Software Engineer - ML Infrastructure

Watney Robotics· Engineering
Apply on company site
San FranciscoFull-timePosted 2mo ago

About the role

ML Infrastructure engineer role focusing on turning robot fleet data into trainable models, building large-scale training/inference infrastructure, data pipelines, and accelerating experiment velocity.

Responsibilities

  • Own training and inference infrastructure.
  • Build the data pipelines that these training runs depend on.
  • Make experiments fast to launch and reproduce.
  • Contribute to our core training code.

Qualifications

  • Have built ML infrastructure that carried real production training runs.
  • Have scaled distributed training systems.
  • Strong experience with Python, PyTorch or TensorFlow.
  • Have experience identifying and troubleshooting GPU performance bottlenecks in large-scale training environments.

Skills

Required:PythonPyTorch or TensorFlowDistributed trainingGPU performance optimization
Last updated Sep 19, 2026. Interested? Apply on the company site.

About Watney Robotics

Watney develops autonomous robotics systems to operate datacenter and supercomputing infrastructure.

Autonomous Physical Infrastructure

autonomous roboticsdatacentersupercomputinginfrastructurehyperscalerscompute scaling
Company profileWebsite

More open roles at Watney Robotics