Waabi · 5 months ago
Senior / Staff ML Optimization Engineer
California, United States
Full-time
Onsite
Senior Level, Lead/Staff
$141K/yr - $249K/yr
6+ years expWaabi is an AI company building the next generation of self-driving technology, founded by AI pioneer Raquel Urtasun. They are seeking a Senior / Staff ML Optimization Engineer to collaborate with autonomy and algorithm engineers to enhance self-driving systems and develop efficient training frameworks.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesMachine Learning
Responsibilities
Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach
Build standardized distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency
Comprehensively profile model runtime and memory to pinpoint performance bottlenecks
Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks, including but not limited to efficient CUDA kernels for training, quantization, model exporting and compilation for inference, etc
Qualification
Required
MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study
Solid coding proficiency in a variety of coding languages including Python, C++ or Rust
Experience in deep learning frameworks such as PyTorch
Experience across different stages of the development pipeline: data processing, distributed training and model deployment
Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight
Open-minded and collaborative team player with willingness to help others
Passionate about self-driving technologies, solving hard problems, and creating innovative solutions
Preferred
Experience in model compilation and exporting, interaction with lower level concepts like TensorRT
Experience in identifying when custom CUDA kernels are needed, and implementing them
Experience in Bazel build systems, and integrating third party packages into dev environments
Benefits
Competitive compensation and equity awards
Health and Wellness benefits encompassing Medical, Dental and Vision coverage
Unlimited Vacation
Flexible hours and Work from Home support
Daily drinks, snacks and catered meals (when in office)
Regularly scheduled team building activities and social events both on-site, off-site & virtually
World-class facility that includes a gym, games room (ping pong table, video game consoles, board games,etc), multiple collaborative working spaces and a gorgeous patio!(when in office)
Company
Waabi
Waabi is pioneering Physical AI starting with autonomous trucking.
51-200 employees
H1B Sponsorship
Waabi has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (6)
2023 (6)
2022 (4)
Funding
Current Stage
Growth StageTotal Funding
$283.04MKey Investors
Volvo Group Venture CapitalKhosla Ventures
2024-06-18Series B· $200.48M
2023-01-18Series Unknown
2021-06-08Series A· $82.55M
Leadership Team
Raquel Urtasun
Founder & CEO
Lior Ron
COO
Recent News
Calgary Herald
2026-01-08
2025-12-13
Company data provided by crunchbase