Senior ML Platform Engineer (Kubernetes & Training Infra)

Waabi; United States; À distance; USD 184 000 - 272 000; Plein temps

Avantages offerts par ce poste

Equity awards

Unlimited vacation

Flexible hours

Work from home support

Catered meals

Résumé du poste

Waabi is building the next era of autonomous transportation with a strong focus on the ML infrastructure that supports training at scale. This role drives the training platform, containerized tooling, and developer experiences for researchers and engineers across a distributed GPU ecosystem.

You will shape APIs/CLIs, implement robust pipelines, and ensure reliable, observable systems that scale with the company’s ambitious autonomy programs.

Qualifications

• 5+ years of software or infrastructure engineering, including ML or data‑intensive production systems.
• Hands-on Kubernetes expertise — GPU scheduling, autoscaling, Helm or equivalent, networking fundamentals, and debugging under load.
• Excellent Python and a track record of designing APIs and CLIs used by others.
• Distributed training in PyTorch (DDP, FSDP) and experience with experiment tracking/model registry tooling.
• Fluency with containers, CI/CD, and modern build systems, including large monorepos.
• Ability to influence without authority; collaborative by default; strong product instincts and writing.

Responsabilités

• Build and evolve training infrastructure on Kubernetes with GPU scheduling, autoscaling, and multi‑node jobs.
• Shape developer-facing surface: CLIs, SDKs, job submission templates for easy use.
• Reduce time to first training run and improve iteration speed across the ML stack.
• Promote best‑in‑class tooling and frameworks with prototypes and migration paths.
• Strengthen data/artifact layers with dataset versioning and high‑throughput loading.

Connaissances

Kubernetes

Python

PyTorch (Distributed)

APIs/CLIs design

CI/CD

Build systems

DevEx tooling

Outils

Argo Workflows

Ray

Flyte

Kubeflow

Slurm

Terraform/Pulumi

NCCL/GPUs networking

Description du poste

Waabi is building the next era of autonomous transportation with a strong focus on the ML infrastructure that supports training at scale. This role drives the training platform, containerized tooling, and developer experiences for researchers and engineers across a distributed GPU ecosystem.

You will shape APIs/CLIs, implement robust pipelines, and ensure reliable, observable systems that scale with the company’s ambitious autonomy programs.