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
Company Information
Waabi is a technology startup dedicated to transforming the way the world moves. Join our talented team to be part of the future and make a significant impact.
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
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.