Senior / Staff ML Training Optimization Engineer
Location
toronto, on
Job Type
Full-time
Category
Other-General
Posted
June 04, 2026
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...
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...