Senior Software Engineer, Machine Learning
Standard Bots$300,000 - $350,000/year
- Full Time
- Senior (5 to 8 years)
Not SpecifiedCompensation
Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level
Full TimeJob Type
UnknownVisa
Autonomous Vehicles, Artificial Intelligence, AI & Machine LearningIndustries
Requirements
Candidates should possess an MS/PhD or a Bachelor’s degree with a minimum of 6 years of industry experience in Computer Science, Robotics, or a related technical field. Strong coding proficiency in Python, C++, or Rust is required, along with experience in deep learning frameworks such as PyTorch and familiarity with various stages of the development pipeline, including data processing, distributed training, and model deployment. Proficiency in profiling CPU and GPU code using tools like PyTorch Profiler and NVIDIA Nsight is also necessary.
Responsibilities
The Senior/Staff ML Optimization Engineer will collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach, build standardized distributed training frameworks, comprehensively profile model runtime and memory to identify performance bottlenecks, and evaluate emerging technologies for training and inference frameworks, including efficient CUDA kernels, quantization, model exporting, and compilation. They will also identify when custom CUDA kernels are needed and implement them, and experience with Bazel build systems and integrating third-party packages into development environments is beneficial.
Skills
Python
C++
Rust
PyTorch
Distributed Training
Profiling Tools
CUDA Kernels
Quantization
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