ML Optimization Engineer

Drive the future of self-driving tech with Waabi as a Senior Machine Learning Optimization Engineer. Utilize your expertise to enhance AI systems, focusing on stability and efficiency.

At Waabi, we are seeking an experienced ML Optimization Engineer with a minimum of six years in Computer Science or Robotics. Collaborate with autonomy teams, build distributed training frameworks, and improve model runtime profiling. Your skills in coding languages like Python or C++, along with deep learning frameworks, will be essential for our projects.

• Scale safe self-driving systems using an AI-first methodology
• Build standardized frameworks for distributed training
• Profile model runtime and memory to identify bottlenecks
• Evaluate emerging technologies for training and inference
• Collaborate with algorithm engineers to enhance model performance

• MS/PhD or Bachelor’s with 6+ years experience
• Proficient in Python, C++, or Rust programming
• Experience with PyTorch deep learning framework
• Skilled in profiling tools like PyTorch Profiler
• Passionate about self-driving technology

Bring your team-oriented spirit and technical skills to Waabi's cutting-edge projects in self-driving technology.