OpeningsFirst
How it worksPricingLog in
OpeningsFirst › Jobs in the United States › Machine Learning / AI Engineer jobs in the United States

Staff Software Engineer - ML Infrastructure

Watney · watney.com
San FranciscoOn-siteLead / Manager
Apply on Watney’s site
Posted 4 months ago. 684 more Machine Learning / AI Engineer jobs in the United States went up on company career pages this week. See the newer ones

Share this job on LinkedIn · X

Straight from the company’s own careers site.

The job in short

A lead / manager full-time on-site Machine Learning / AI Engineer role at Watney in San Francisco. Skills mentioned: Python, PyTorch, TensorFlow.

PythonPyTorchTensorFlow

Details

Found
38 d ago · posted 2026-07-13
Location
San Francisco

Full description

Our Mission Expand human ambition in the physical world. Critical infrastructure is constrained by labor shortages, hazardous working conditions, and operational complexity. Watney builds and deploys autonomous robotic systems that increase the speed and capacity of buildout, starting with data centers. About the Role At Watney, ML Infrastructure engineers turn data collected from a live fleet of robots into better models. The fleet produces large volumes of video and telemetry data from real work in the field, and making that data trainable is one of the hardest systems problems at the company. As we continue to scale, these systems will require larger training runs with more data, expanded clusters, and optimal GPU utilization. What You’ll Do • 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 . You May Be a Good Fit If You: • 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 . We’re committed to building a diverse, inclusive team. At Watney Robotics, we welcome people of all backgrounds and identities, and we make hiring decisions based on skills, experience, and potential. If you’re passionate about robotics but don’t meet every requirement, we still encourage you to apply! Curious to learn more? Follow us here on X and LinkedIn

All Watney jobs · Senior Machine Learning / AI Engineer jobs in the United States

Get jobs like this every morning

Jobs in India · Jobs in the US · Who’s hiring in India · Who’s hiring in the US · Browse all jobs · Companies · How OpeningsFirst works · Free jobs widget · Leadership jobs in India · Leadership jobs in the US · Senior jobs in India · Senior jobs in the US · Fresher jobs in India · Entry-level jobs in the US · Remote jobs in the US · Jobs paying $150k+ · Leadership hiring this week · Guides
OpeningsFirst · Use with AI · Terms · Privacy · Refunds · Contact · Follow us on LinkedIn · X