Senior / Staff ML Ops Engineer
WaabiVisit Waabi website
Waabi is a Physical AI company developing autonomous-driving technology for commercial trucking and robotaxis.
Toronto, Canada
About Waabi
Founded in 2021 by Raquel Urtasun, Waabi builds a Physical AI platform combining a verifiable end-to-end AI model with neural simulation. Its Waabi Driver and Waabi World technologies support autonomous trucking and robotaxi development.
Skills
About the Role
You will build and evolve training infrastructure on Kubernetes, shape developer-facing tools like CLIs and SDKs, shorten the inner loop for researchers, strengthen the data and artifact layer, and ship CI/CD and observability across the ML stack. You will also drive adoption through docs, onboarding and support, and build self-serve guardrails.
Requirements
- 5+ years of software or infrastructure engineering, including tools or platforms for engineers and ML or data-intensive production systems
- Excellent Python and API/CLI design
- Practical AWS depth including storage, IAM, GPU compute, networking, cost management and infrastructure as code
- Distributed training in PyTorch (DDP, FSDP), experiment tracking and model registry tooling
- Fluency with containers, CI/CD and modern build systems including monorepos
- Ability to influence without authority
- Strong product instincts and writing
Responsibilities
- Build and evolve training infrastructure on Kubernetes including GPU scheduling, autoscaling and distributed jobs
- Shape developer-facing CLIs, SDKs, job submission and templates
- Shorten the inner loop for training runs and measure it
- Evaluate and evangelize best-in-class tooling and frameworks
- Strengthen dataset versioning, sharding and high-throughput loading of multimodal sensor data
- Turn one-off Python into durable tested tooling
- Make experiments legible with experiment tracking, dashboards and a model registry
- Ship CI/CD for models alongside autonomy and simulation
- Build observability across the ML stack
- Treat docs, onboarding and support as product surface
- Drive adoption of tools with real users
- Make the platform reliable
- Build self-serve guardrails for access control, data handling and cost governance
Benefits
- Competitive compensation and equity awards
- Medical, Dental and Vision coverage (full-time employees)
- Unlimited Vacation
- Flexible hours and Work from Home support
- Daily drinks, snacks and catered meals (when in office)
- Team building activities and social events
- Annual performance bonus
