# Senior / Staff ML Training Optimization Engineer at Waabi

- Company: Waabi
- What the company does: Transform your supply chain with Waabi’s AI-first autonomous trucking solution. Scale freight operations, improve safety, and boost efficiency. Learn more. Backed by Khosla.
- Company website: https://waabi.ai/
- Type: Startups (AI role)
- Level: Senior
- Location: Remote US & Canada
- Work setup: Hybrid
- Pay: $141K to $249K base salary per year (USD)
- Posted: 2026-05-08
- Apply by: 2026-10-08
- Apply: https://jobs.lever.co/waabi/eedaf111-0efd-4f8f-9c2d-ce829b5cb0eb
- Page: https://www.1752.vc/careers/jobs/waabi-senior-staff-ml-training-optimization-engineer/

## About the role

- Create and improve tooling and dashboards to ensure broad adoption of your work.

## What they're looking for

- - MS/PhD or Bachelors degree with a minimum of 4 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 or Jax
- - 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

Tags: Autonomy & Algorithms