PnP Optimization Engineer

San Francisco, California, United States

Apply with AI Apply

$158,000 – $269,000Compensation

Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level

Full TimeJob Type

UnknownVisa

Artificial Intelligence, Autonomous Vehicles, BiotechnologyIndustries

Job Description: Machine Learning Engineer

Position Overview

Waabi is an AI company at the forefront of developing the next generation of self-driving technology. Founded by AI pioneer Raquel Urtasun, Waabi leverages a world-class team and an innovative approach to safely drive in real-world conditions, accelerating the commercialization of self-driving technology. Backed by leading investors in technology, logistics, and the Canadian innovation ecosystem, Waabi is rapidly expanding with offices in Toronto, San Francisco, and Dallas. We are seeking diverse, innovative, and collaborative candidates eager to make a positive global impact.

Employment Type

Full-time

Salary

US Yearly Salary Range: $158,000 - $269,000 USD
Additional Compensation: Competitive perks & benefits, equity incentive awards, and an annual performance bonus.
Note: Salary ranges are determined based on various factors and reflect target new hire salaries across all US locations.

Perks & Benefits

Competitive compensation and equity awards
Health and Wellness benefits (Medical, Dental, Vision) for full-time employees
Unlimited Vacation
Flexible hours and Work From Home support
Daily drinks, snacks, and catered meals (when in office)
Regularly scheduled team-building activities and social events (on-site, off-site, and virtual)
This list is evolving as we grow.

Responsibilities

Collaborate with research scientists during algorithm design to ensure efficient code development from inception.
Identify and communicate best practices for model development.
Reformulate performance bottlenecks.
Develop a framework suitable for mixed precision training of multi-task, multi-modality models in a heterogeneous distributed training environment.
Research, implement, and test alternative formulations for fundamental DNN operations and AV-centric representations.
Enable safe and efficient deployment of PnP models.

Qualifications

PyTorch API and implementation
CUDA system design
Asynchronous programming model
Heterogeneous compute
C++ / libtorch

Bonus Qualifications

Machine learning model architecture design
TensorRT
CUDA kernel implementation
Experience with autonomous vehicle multi-sensor model architectures

Application Instructions

Visit www.waabi.ai to learn more.

Equal Opportunity Employer

Waabi is committed to celebrating diversity and creating a supportive, inclusive, and accessible workplace for all employees. We welcome applicants from all backgrounds and identities, including but not limited to race, color, ethnicity, national origin or ancestry, age, citizenship, religion, sex, sexual orientation, gender identity or expression, military or veteran status, marital status, pregnancy or parental status, caregiver status, disability, or any other characteristic protected by law.

Waabi provides workplace accommodations for qualified individuals with disabilities as required by applicable law. If reasonable accommodation is needed to participate in the job application or interview process, please contact our recruiting team.

Skills

PyTorch API

CUDA system design

Asynchronous programming

Heterogeneous compute

C++

libtorch

Machine learning model architecture design

TensorRT

CUDA kernel implementation

Autonomous vehicle multi-sensor model architectures

Waabi

Develops self-driving technology for trucking

About Waabi

Waabi focuses on developing self-driving technology specifically for the trucking industry. Their main product, the Waabi Driver, is designed to learn from experiences, adapt to new situations, and work with various hardware setups. This technology aims for large-scale use and is intended for integration into vehicles by original equipment manufacturers (OEMs). A key feature that sets Waabi apart from competitors is its use of Waabi World, an advanced simulation environment that allows for efficient development of their self-driving systems. This reduces the need for extensive real-world testing, which can be costly and time-consuming, while also improving safety and adaptability. Waabi's goal is to transform the trucking industry by providing technology that lowers operational costs and enhances safety for commercial vehicles. They aim to license their technology to trucking companies and OEMs, making it easier for these businesses to adopt self-driving solutions.

Toronto, CanadaHeadquarters

2021Year Founded

$265.9MTotal Funding

SERIES_BCompany Stage

Automotive & Transportation, AI & Machine LearningIndustries

51-200Employees