πŸ‡ΊπŸ‡Έ USAJobs.work

America's Job Portal

← Back to USA Jobs

Senior / Staff ML Training Optimization Engineer

Company

Waabi

Location

toronto, on

Posted

June 05, 2026

Position Overview

Requirements

  • 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
  • (Desirable) Experience in identifying when custom CUDA kernels are needed, and implementing them
  • (Desirable) Experience in Bazel in a monorepo environment, and integrating third party packages into dev environments
  • (Desirable) Experience with Kubernetes-based training platforms...

Ready to Apply?

Join thousands of Americans building their careers

Apply Now