Build Kubernetes training infrastructure at Waabi, using Python and AWS to support Physical AI for autonomous trucks and robotaxis.This senior engineering role develops platforms used by ML researchers and engineers. You will shape APIs, CLIs, experiment workflows, and reliable production systems. The work spans distributed PyTorch training, GPU scheduling, data pipelines, and model delivery.Key Responsibilities:
Build Kubernetes infrastructure for distributed GPU training workloadsDesign developer-facing CLIs, SDKs, templates, and workflowsStrengthen dataset versioning, sharding, and artifact loadingImplement observability for utilization, failures, queues, and costsShip CI/CD for models alongside autonomy and simulationRequirements:
Five or more years in software or infrastructure engineeringExcellent Python and experience designing usable APIs and CLIsPractical AWS, IAM, GPU, networking, and infrastructure-as-code expertiseDistributed PyTorch training with experiment tracking and model registriesFluency with containers, CI/CD, and contemporary build systemsUse Python, AWS, Kubernetes, and distributed training expertise to advance Waabi's autonomous transportation infrastructure.