Senior Machine Learning Engineer - Machine Learning Infrastructure
FlipFull Time
Senior (5 to 8 years)
Key technologies and capabilities for this role
Common questions about this position
The position is hybrid in Toronto, San Francisco, or Dallas, with flexible hours and work from home support.
Required skills include solid coding in Python, C++, or Rust; experience with PyTorch and other deep learning frameworks; development pipeline experience in data processing, distributed training, and model deployment; and profiling skills using tools like PyTorch Profiler and NVIDIA Nsight.
Benefits include competitive compensation and equity, health and wellness coverage for medical, dental, and vision, unlimited vacation, flexible hours with work from home support, daily drinks, snacks, and catered meals in office, team building activities, and world-class facilities with a gym and games room.
Waabi fosters a collaborative, innovative environment with diverse team players, regular team building activities, social events, and perks like a games room and gym to support work-life balance.
Strong candidates have an MS/PhD or Bachelor's with 6+ years experience in CS, Robotics or related fields, solid coding skills, deep learning framework experience, and a passion for self-driving tech; bonus for model compilation, custom CUDA kernels, and Bazel.
Develops self-driving technology for trucking
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.