Staff Machine Learning Engineer

The role

ML engineers turn a live fleet of robots into world class models. You will run training experiments, curate and label data, and evaluate performance on a real fleet.

What you will do

• Own model quality end-to-end including dataset curation, training, and evaluation.
• Work with multimodal sensor data from Cameras, IMUs, LiDar and Odometry.
• Train and evaluate policies and experiment with data, feature and architecture ablations.
• Deploy results to production and build systems for low latency, 24/7 inference on a fleet of robots.
• Work with Teleoperations and Fleet Operations to get the data the models need.

Required qualifications

• Trained large scale multimodal models on a fleet of GPUs using techniques such as video semantic segmentation, 6d pose estimation, VLAs and BEVs.
• Have owned a dataset end-to-end: what to collect, what to label, and what to cut.
• Write production training and inference code with Python, PyTorch or JAX, Triton.
• Have found and fixed a problem with a training run or a model's real-world performance.
• Experience working on Cluster management systems including Ray, Slurm or Kubernetes.

About the company

Watney builds and deploys autonomous robotic systems that increase the speed and capacity of buildout, starting with data centers.