Machine Learning Engineer

Waabi, founded by AI pioneer Raquel Urtasun, is an AI company building the next generation of self-driving technology. With a world-class team and an innovative approach that unleashes the power of AI to "drive" safely in the real world, Waabi is bringing the promise of self-driving closer to commercialization than ever before. Waabi is backed by best-in-class investors across the technology, logistics and the Canadian innovation ecosystem.

As a Machine Learning Engineer at Waabi, you will:

• Contribute and build self-driving engineering foundations for next-generation autonomy software across many machine learning projects.
• Collaborate with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach.
• Build standardized frameworks for training and production aligned with industry best practices.
• Profile model runtime and memory to identify performance bottlenecks.
• Evaluate emerging technologies for adoption in Waabi's training and production frameworks.
• Work with AWS Cloud infrastructure to ensure training job stability and efficiency.

Qualifications:

• MS/PhD or Bachelor's degree with 2+ years of industry experience in Computer Science, Robotics, or similar fields.
• Solid coding proficiency in Python, C++, or Rust.
• Experience with deep learning frameworks like PyTorch.
• Experience in distributed machine learning training, model deployment, and monitoring.
• Experience building software solutions on cloud infrastructure.

Waabi offers competitive compensation, equity awards, health benefits, unlimited vacation, flexible hours, and a collaborative work environment. Join a diverse and innovative team working to impact the world positively through self-driving technology.