PitchMeAI Jobs
Waabi

Senior / Staff ML Training Optimization Engineer

Waabi · Remote US & Canada

Posted about 3 months ago · $141,000 - $249,000

or apply directly on Waabi's site. We never take the application ourselves.

Is this posting real?

This role has been open
79 days
Waabi's roles stay open a median of 80 days
Reposted
No
Salary listed
Yes
65% of Waabi's roles list one
Ghost-job risk at Waabi
high
81 stale, 2 reposted of 88 open
Hiring momentum
97 roles opened in the last 90 days
↑ up vs. the prior 90 days
Last confirmed on the employer's board
2026-10-23

Measured from postings appearing on and disappearing from Waabi's own lever board since 2026-08-04. Full hiring picture for Waabi.

About this role

As a Senior / Staff ML Training Optimization Engineer at Waabi, you will focus on building and improving distributed training frameworks for both research and production environments. Your role involves profiling model performance, identifying bottlenecks, and collaborating with researchers and ML engineers to optimize resource usage. You will also create tools and dashboards to enhance the adoption of your work within the team.

Our read on this posting3.6out of 5
benefits
4/5
freshness
1/5
career value
4/5
role clarity
4/5
pay transparency
5/5

Scored from the posting itself — how clearly the role is described, how much it says about pay and benefits, and how recently it was listed. Not a judgement of Waabi as an employer.

What you need

  • MS/PhD or Bachelor's degree with a minimum of 4 years of industry experience in Computer Science, Robotics, or similar technical fields
  • Solid coding proficiency in 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
  • Passionate about self-driving technologies and solving hard problems

Nice to have

  • Experience in identifying when custom CUDA kernels are needed and implementing them
  • Experience in Bazel in a monorepo environment and integrating third party packages into dev environments
  • Experience with Kubernetes-based training platforms

What you get

  • 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

Worth weighing

  • Remote work may limit in-office perks and social interactions
  • Salary range provided is broad, with significant variability based on experience and negotiation
  • Focus on deep learning and CUDA may not appeal to those looking for broader ML applications

Summarised from Waabi's posting. Read the full original.

Listed by Waabi on their lever job board, last confirmed open on 2026-10-23. PitchMeAI is not the employer.

More roles at Waabi

All 88 open roles at Waabi →