Senior ML Platform Engineer - Remote, High-Impact Infra

Waabi; Dallas (TX); On-site; USD 184,000 - 272,000; Full time

Benefits offered by this job

Competitive compensation and equity

Health and Wellness benefits (medical,

Unlimited Vacation

Flexible hours and Work from Home

Daily drinks, snacks and meals

Team building activities

Evolving benefits

Job summary

Waabi is building next-gen autonomous systems infrastructure and tooling in Dallas, TX. We are hiring engineers to design and maintain scalable training infrastructure, developer-facing APIs, and reliable pipelines that power large-scale experiments.

You will implement Kubernetes-based training workloads, streamline tooling, and collaborate with researchers and engineers to deliver robust model training platforms. This role offers substantial impact and growth within a world-class AI team.

Qualifications

• 5+ years of software or infrastructure engineering, including tools or platforms used by other engineers and operating ML or data-intensive production systems.
• Hands-on Kubernetes expertise — GPU scheduling, autoscaling, Helm or equivalent, networking fundamentals, and the ability to debug a cluster under load rather than restart it.

Responsibilities

• Build and evolve our training infrastructure on Kubernetes with Infrastructure — GPU scheduling, autoscaling, multi-node distributed jobs, capacity strategy, and the operators and workflow engines that keep long-running training reliable.
• Shape the developer-facing surface — CLIs, SDKs, job submission, templates, paved paths — designed with the teams who'll use them.
• Shorten the inner loop. Time to first training run, edit-to-signal latency, local iteration before a job hits the cluster, fast failure over slow mystery.
• Evangelize best-in-class tooling and frameworks. Track what the ecosystem is shipping, evaluate honestly, and make the case with working prototypes and migration paths.
• Strengthen the data and artifact layer. Dataset versioning, sharding, and high-throughput loading of large multimodal sensor data.
• Turn one-off Python into durable tooling — tested, documented, observable libraries, CLIs, and services with sane defaults.

Skills

Kubernetes

Python

APIs & CLIs

Distributed training

CI/CD

AWS

Team collaboration

Debugging under load

GPU scheduling

Tools

Terraform

Pulumi

Docker

Kubernetes

NCCL

Job description

Waabi is building next-gen autonomous systems infrastructure and tooling in Dallas, TX. We are hiring engineers to design and maintain scalable training infrastructure, developer-facing APIs, and reliable pipelines that power large-scale experiments.

You will implement Kubernetes-based training workloads, streamline tooling, and collaborate with researchers and engineers to deliver robust model training platforms. This role offers substantial impact and growth within a world-class AI team.