Senior / Staff ML Optimization Engineer

Join Waabi, an AI company building self-driving technology, as a Senior/Staff ML Optimization Engineer. Collaborate with a world-class team to scale safe self-driving systems and drive innovation in AI.

Requirements

• MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study
• Solid coding proficiency in a variety of coding languages including Python, C++ or Rust
• Experience in deep learning frameworks such as PyTorch
• Experience across different stages of the development pipeline: data processing, distributed training and model deployment
• Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight
• Open-minded and collaborative team player with willingness to help others
• Passionate about self-driving technologies, solving hard problems, and creating innovative solutions

Benefits

• Competitive compensation and equity awards
• Health and Wellness benefits encompassing Medical, Dental and Vision coverage
• 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 both on-site, off-site & virtually
• World-class facility that includes a gym, games room (ping pong table, video game consoles, board games,etc), multiple collaborative working spaces and a gorgeous patio!(when in office)