Latest AI JobsBrowse Jobs
W

Staff Machine Learning Engineer

watneySan FranciscoEngineering
Apply on company site
Salary
Not listed
Job type
Full-time
Experience
Senior
Posted
16 Sept 2026
Status
Open · checked 7d ago

About this role

This position suits an experienced machine learning engineer based in San Francisco who wants to optimize autonomous robotic systems for data centers. The Staff Machine Learning Engineer role at watney involves transforming field video data into high-performing production models for a live fleet of robots.

Role Overview

Watney seeks a Staff Machine Learning Engineer in San Francisco to manage model quality and improve robot performance in physical operations. The team develops autonomous systems to address labor shortages and operational complexity in critical infrastructure.

Key Responsibilities

Supervise model quality across dataset curation, training, and evaluation.
Process multimodal sensor data from cameras, IMUs, LiDAR, and odometry for training sets.
Train and test policies while conducting ablations on data, features, and architectures.
Push high-performing models to production and maintain low-latency inference systems for the robot fleet.
Coordinate with teleoperations and fleet operations to secure necessary training data.

Requirements

Experience training large-scale multimodal models on GPU clusters using video semantic segmentation, 6D pose estimation, VLAs, and BEVs.
Background in managing datasets end-to-end, including collection, labeling, and pruning.
Proficiency in writing production code using Python, PyTorch, JAX, or Triton.
Track record of diagnosing and resolving issues in training runs or real-world model deployment.
Familiarity with cluster management systems such as Ray, Slurm, or Kubernetes.

Skills and Tools

Python
PyTorch
JAX
Triton
Ray
Slurm
Kubernetes
Cameras, IMUs, LiDAR, and Odometry sensors

About the Company

Watney builds and deploys autonomous robotic systems aimed at increasing the speed and capacity of infrastructure buildouts, starting with data centers. The organization maintains a commitment to an inclusive team environment and bases hiring choices on skills, experience, and potential.

How to Apply

Interested candidates should visit the official company careers page to submit an application.

Skills and tech stack

PythonPyTorchJAXTritonRaySlurmKubernetesMachine Learning

Questions candidates ask

Where is the job located?

The role is located in San Francisco.

What kind of systems does the company build?

Watney builds autonomous robotic systems focused on data centers and physical infrastructure buildouts.

What programming languages and frameworks are required?

The position requires writing production training and inference code using Python, PyTorch, JAX, or Triton.

What sensor data will the engineer work with?

The engineer will work with multimodal sensor data from cameras, IMUs, LiDAR, and odometry.

Are there specific cluster management tools mentioned?

Yes, experience working on cluster management systems including Ray, Slurm, or Kubernetes is required.

Ready to apply?

The application is on the official watney careers page.

Apply on company site