← Back to opportunities

Senior / Staff ML Training Optimization Engineer

📍 Location
toronto
⏰ Job Type
Full-time
📅 Posted
June 04, 2026

About the Role

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

What the job involves

Build standardized d...

Ready to Join Through a Referral?

Apply now and get connected directly with the hiring team

Apply for this Position