Staff Software Engineer - ML Infrastructure

Posted 5 Days Ago
Be an Early Applicant
San Francisco, CA, USA
In-Office
Senior level
Artificial Intelligence • Information Technology • Robotics • Software
The Role
Design, own, and scale training and inference infrastructure for a robotic fleet. Build data pipelines, optimize distributed training and GPU utilization, speed experiment launch and reproducibility, and contribute to core training code.
Summary Generated by Built In
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

Skills Required

  • Built ML infrastructure that carried real production training runs
  • Experience scaling distributed training systems
  • Strong experience with Python
  • Strong experience with PyTorch or TensorFlow
  • Experience identifying and troubleshooting GPU performance bottlenecks in large-scale training environments
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, California
13 Employees
Year Founded: 2024

What We Do

Join us on our mission to deploy an abundant, intelligent workforce. Watney develops autonomous robots to enable the growth of critical infrastructure. Having recently raised $21 million in seed funding from Conviction, Abstract and A*, we’re partnering with the world’s largest hyperscalers to accelerate datacenter buildout and breakfix. You'll work alongside - The former GSM for the iPhone - The former founding engineer at Worldcoin - The engineer who built the world’s fastest EV racecar at ETH Zurich - … and many more Join now for a ground-floor seat with outsized ownership and visibility as we scale from prototype to production fleets. Ship real systems, shape the rollout playbook, and leave fingerprints on a category-defining robotics company.

Similar Jobs

Nuro Logo Nuro

Staff Software Engineer

Artificial Intelligence • Automotive • Information Technology • Robotics
In-Office
Mountain View, CA, USA
908 Employees
194K-352K Annually

Voxel Logo Voxel

Staff Software Engineer

Artificial Intelligence • Security • Software
Hybrid
San Francisco, CA, USA
62 Employees
220K-260K Annually

Nuro Logo Nuro

Staff Software Engineer

Artificial Intelligence • Automotive • Information Technology • Robotics
In-Office
Mountain View, CA, USA
908 Employees
194K-352K Annually
In-Office
2 Locations
2359 Employees
251K-310K Annually

Similar Companies Hiring

Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
LTX Thumbnail
Robotics • Conversational AI • Generative AI
Jerusalem, Israel
200 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account