Salary
≈ $223k – $420k per year (Estimated)
Location
Remote (United States, Canada)
Seniority
Staff · 5+ years exp
Visa
H-1B filings in 12 months: 9 · for this role: 6
Employment
Full-Time
Confirmed on the employer's own hiring board on Nov 11, 2026. First seen by Alion on Sep 23, 2026. Waabi scores A on the Alion truth index.
Overview
Company
Impact
Profile match
Waabi is an artificial intelligence and autonomous vehicle technology company headquartered in Toronto, Canada, founded in 2021. The company develops an end-to-end artificial intelligence software platform, known as the Waabi Driver, and a simulation environment called Waabi World to power autonomous trucks and robotaxis. It operates primarily in the North American logistics and transportation sectors through strategic partnerships with industry leaders such as Volvo, Uber Freight, and NVIDIA.
Backed by Khosla Ventures, 8VC, Radical Ventures
You will..
- 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. Make the common case one command and keep the uncommon case possible.
- 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. Measure it, publish it, drive it down.
- 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 - or say plainly when a shiny thing isn't worth the switching cost.
- Strengthen the data and artifact layer. Dataset versioning, sharding, and high-throughput loading of large multimodal sensor data, so jobs saturate GPUs instead of waiting on I/O.
- Turn one-off Python into durable tooling - tested, documented, observable libraries, CLIs, and services with sane defaults, and deletions where they're overdue.
- Make experiments legible, with the teams who live in them: experiment hygiene, dashboards researchers trust, a real model registry, and lineage from dataset to checkpoint to simulation result.
- Ship CI/CD for models alongside autonomy and simulation, so a model change is validated the same way a code change is.
- Build observability across the ML stack - utilization, throughput, failure modes, queue times, cost per experiment. When a job fails at 3am on node 47, the researcher should find out why without you.
- Treat docs, onboarding, and support as product surface - golden-path guides, a new researcher productive on day two, office hours that turn repeat questions into shipped fixes.
- Drive adoption, not just availability. Prototype with real users, watch them work, iterate. A tool nobody adopts didn't ship.
- Make the platform boringly reliable - fewer failures, faster recovery, and none of the manual steps that quietly cost a team days.
- Build guardrails that don't feel like walls, with Security, IT, and Infrastructure: access controls, data handling, and cost governance that hold up in an IP-sensitive environment while staying self-serve.
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.
- Excellent Python, and a track record of designing APIs and CLIs other people enjoy using.
Practical AWS depth: object storage at scale, IAM, GPU compute, networking, cost management, and infrastructure as code (Terraform, Pulumi, or similar).
- Distributed training in PyTorch (DDP, FSDP, or similar), plus experiment tracking and model registry tooling - from the perspective of someone who made them pleasant for others to use.
- Fluency with containers, CI/CD, and modern build systems, including large monorepos.
- The ability to influence without authority: evaluate a framework on its merits, pilot it credibly, and persuade skeptical senior engineers to change how they work.
- A collaborative default - you'd rather co-own a system than draw a boundary around your part of it.
- User empathy: you'd rather fix the third-most-interesting problem blocking ten people than the most interesting one blocking nobody.
- Strong product instincts, strong writing, and comfort operating autonomously in ambiguous territory.
- Passionate about self-driving technologies and frontier AI, and about what a small, world-class team can do with the right infrastructure.
Bonus/nice to have:
- Internal developer platform, research platform, or DevEx work - with a story about a tool whose adoption you grew from zero.
- Large-scale distributed GPU training: hundreds to thousands of accelerators, NCCL, high-performance cluster networking, collective communication tuning.
- High-throughput loading of LiDAR or camera data, and formats such as Parquet or WebDataset.
- Workflow and scheduling systems - Argo Workflows, Ray, Flyte, Kubeflow, or Slurm.
- Build-system depth (Bazel or similar), including remote caching in a monorepo.
- Simulation infrastructure or large-scale batch evaluation pipelines.
- Background in ML, robotics, or autonomous systems infrastructure.
- Security- and IP-sensitive production environments.
- Open-source contributions to ML infrastructure or developer tools.
Free account
Stop reading job ads. Get the ones that fit.
One free account turns this page into a shortlist built around your stack, your level and your pay.
Match on every job. Stack, seniority, pay and location, scored against your profile.
2,200,306 open roles. Read straight off company career pages, refreshed every day.
Unlimited applications. Every one you send is tracked in one place, on-site or on a company board.
3 tailored CVs a month. Rewritten for the exact job you are applying to. Included free.
Free forever. No card. Under a minute.
Your match
How well do you fit this role?
Two answers are enough for a real match. No account needed.
Check my fit
Answers stay in this browser until you create an account.
Recommended for you based on this role
AI/ML
Similar stack
Same company
San Francisco
Senior AI Software Engineer - Integrations
1 hour ago
Remote (likely Peru) · 8+ years exp
Python
JavaScript
Rust
TypeScript
Node JS
Node JS
Nest.JS
Databases
PostgreSQL
DevOps
Rest API
AWS
Kubernetes
Cybersecurity
PCI DSS
Apply
Remote (likely Peru) · 8+ years exp
Python
JavaScript
Rust
TypeScript
Node JS
Node JS
Nest.JS
Databases
PostgreSQL
DevOps
OpenTelemetry
AWS
Kubernetes
QA
Sentry
Apply
Staff Machine Learning Scientist
2 hours ago
$134k – $235k per year · Equity · In office · 8+ years exp · Bachelor's Degree · Santa Clara
Python
AI/ML
NLP
TensorFlow
LLM
CUDA
Machine Learning
DevOps
GCP
Azure
AWS
Cybersecurity
Zero Trust
Apply
Senior AI & Automation Engineer
5 hours ago
$200k – $230k per year · Hybrid · 8+ years exp · New York
PowerShell
AI/ML
LangChain
Function Calling
AI Agents
Semantic Kernel
RAG
Hallucination
LLM Guardrails
Copilot Studio
DevOps
Azure
CI/CD
Management
ServiceNow
Power Automate
Power Apps
Agile
ITIL
Apply
AI Scientist (LATAM)
6 hours ago
Remote (EEA, Europe, Brazil, PST hours) · Full-Time · 5+ years exp · São Paulo
Python
SQL
Databases
Google BigQuery
AI/ML
Cursor
Spark
Claude Code
Fine-tuning
AI Agents
Machine Learning
DevOps
Prometheus
Cortex
Web3
TRM Labs
Management
Notion
Apply
Eng I, Eng Run
16 days ago
Equity · Hybrid · Full-Time · Bachelor's Degree · Hyderabad
Python
SQL
Databases
Snowflake
DevOps
AWS
Analytics
ETL/ELT
Informatica
Management
Agile
Scrum
Apply
Sr. DevOps Engineer
16 days ago
≈ $77k – $148k per year (Estimated) · Remote (United Kingdom) · Full-Time · United Kingdom
SQL
Databases
Redis
Presto
Apache Kafka
AI/ML
Hadoop
DevOps
Terraform
Azure
CI/CD
Kubernetes
Linux
Windows
Apply
STAGE - Ingénieure / Ingénieur en Amélioration des méthodes d'exploitation de données d'essais
16 days ago
≈ $22k – $41k per year (Estimated) · In office · Internship · Vernon
Python
AI/ML
Machine Learning
DevOps
Prometheus
Apply
In office · Internship · Les Mureaux
Python
Apply
STAGE - Ingénieure / Ingénieur Navigation pour un véhicule de rentrée atmosphérique sur Mars
2 days ago
≈ $21k – $40k per year (Estimated) · In office · Internship · Les Mureaux
Python
MATLAB
Apply
≈ $221k – $417k per year (Estimated) · Remote (United States, Canada) · Full-Time · 4+ years exp · Bachelor's Degree · San Francisco · Dallas · Pittsburgh · Phoenix · Toronto
Python
C++
AI/ML
PyTorch
Machine Learning
Apply
Senior / Staff ML Training Optimization Engineer
6 months ago
≈ $228k – $430k per year (Estimated) · Remote (United States, Canada) · Full-Time · 4+ years exp · Bachelor's Degree · San Francisco · Dallas · Pittsburgh · Phoenix · Toronto
Python
Rust
C++
AI/ML
Quantization
PyTorch
CUDA
DevOps
Kubernetes
Bazel
Apply
Research Scientist, Simulation Agents
8 months ago
≈ $184k – $409k per year (Estimated) · Remote (United States, Canada) · Full-Time · Master's Degree · San Francisco · Dallas · Pittsburgh · Phoenix · Toronto
AI/ML
Reinforcement Learning
TensorFlow
PyTorch
Machine Learning
Robotics
Imitation Learning
Reinforcement Learning
Apply
≈ $227k – $429k per year (Estimated) · Remote (United States, Canada) · Full-Time · 6+ years exp · Bachelor's Degree · San Francisco · Dallas · Pittsburgh · Phoenix · Toronto
Python
Rust
C++
AI/ML
TensorRT
PyTorch
CUDA
Apply
Research Engineer
2 years ago
≈ $172k – $384k per year (Estimated) · Remote (United States, Canada) · Full-Time · Bachelor's Degree · San Francisco · Dallas · Pittsburgh · Phoenix · Toronto
Python
Rust
C++
AI/ML
Computer Vision
PyTorch
CUDA
Machine Learning
Apply
Digital Innovation and AI Developer
1 day ago
$140k – $155k per year · Remote (United States) · Full-Time · 5+ years exp · Bachelor's Degree · Los Angeles · Costa Mesa · San Francisco · Cleveland · Philadelphia
AI/ML
AI Agents
Copilot Studio
DevOps
Rest API
Azure DevOps
Azure
Git
Analytics
Power BI
Management
Miro
Power Automate
Power Apps
SharePoint
Agile
QA
Swagger
Apply
Apply
SIU Field Investigator - Sacramento, CA
11 hours ago
$60k – $87k per year · Hybrid · Full-Time · PhD · San Francisco · Sacramento · Oakland · San Jose
Apply
Director, Banking Partnerships - Americas
12 hours ago
$220k – $330k per year · Hybrid · Full-Time · San Francisco · New York
Apply
Principal Infrastructure Security Engineer
12 hours ago
$220k – $240k per year · In office · San Francisco
DevOps
CI/CD
AWS
Kubernetes
IAM
Cybersecurity
ISO 27001
SOC 2
Least Privilege
Apply
This is one of many
2,200,306 more open roles from verified company boards, updated every day.

