Taro Logo

PnP Optimization Engineer

AI company building next generation self-driving technology, founded by AI pioneer Raquel Urtasun.
$158,000 - $269,000
Machine Learning
Mid-Level Software Engineer
Hybrid
AI · Automotive

Description For PnP Optimization Engineer

Waabi, an innovative AI company founded by AI pioneer Raquel Urtasun, is revolutionizing self-driving technology. They're seeking a PnP Optimization Engineer to join their world-class team across offices in Toronto, San Francisco, Dallas, and Pittsburgh. This role combines machine learning expertise with performance optimization, requiring deep knowledge of PyTorch, CUDA, and C++.

The position involves collaborating with research scientists to optimize algorithm design, develop frameworks for mixed precision training, and ensure efficient model deployment. The ideal candidate will work on cutting-edge autonomous vehicle technology, focusing on performance optimization and implementation of deep neural network operations.

This is an excellent opportunity for someone passionate about AI and autonomous vehicles, offering a competitive salary range of $158,000 - $269,000 USD, plus equity and benefits. The company provides a flexible hybrid work environment with offices in major tech hubs, comprehensive health benefits, unlimited vacation, and a strong focus on team building and workplace culture.

Waabi stands out for its innovative approach to self-driving technology and its commitment to diversity and inclusion. They're backed by top-tier investors and are rapidly growing their team of talented professionals who want to make a positive impact on the world through revolutionary autonomous vehicle technology.

Last updated 19 days ago

Responsibilities For PnP Optimization Engineer

  • Collaborate with research scientists during algorithm design for efficient code design
  • Identify and communicate best practices for model development
  • Reformulate performance bottlenecks
  • Develop framework for mixed precision training of multi-task, multi-modality models
  • Research, implement, and test alternative formulations for DNN operations
  • Enable safe and efficient deployment of PnP models

Requirements For PnP Optimization Engineer

Python
  • PyTorch API and implementation experience
  • CUDA system design knowledge
  • Asynchronous programming model expertise
  • Heterogeneous compute experience
  • C++ / libtorch proficiency

Benefits For PnP Optimization Engineer

Medical Insurance
Dental Insurance
Vision Insurance
Equity
  • Competitive compensation and equity awards
  • Health, Dental and Vision coverage
  • Unlimited Vacation
  • Flexible hours and Work from Home support
  • Daily drinks, snacks and catered meals
  • Regular team building activities and social events

Jobs Related To Waabi PnP Optimization Engineer