Senior / Staff ML Optimization Engineer - Remote

Overview

Waabi is an AI company focused on developing the next generation of self-driving technology, seeking a Senior / Staff ML Optimization Engineer to enhance their training frameworks and improve model performance.

In Short

• Collaborate with engineers to scale self-driving systems.
• Build distributed training frameworks for research and production.
• Profile model runtime and memory to identify bottlenecks.
• Evaluate emerging technologies for training and inference frameworks.
• Work with various coding languages including Python, C++, and Rust.
• Experience in deep learning frameworks like PyTorch.
• Participate in all stages of the development pipeline.
• Skilled in profiling CPU and GPU code.
• Collaborative team player passionate about self-driving technologies.
• Possibility to work with custom CUDA kernels and Bazel build systems.

Requirements

• MS/PhD or Bachelor's degree with 6+ years in a technical field.
• Solid coding proficiency in Python, C++, or Rust.
• Experience with deep learning frameworks like PyTorch.
• Knowledge of data processing and model deployment.
• Ability to profile code using tools like PyTorch Profiler.
• Open-minded and collaborative team player.
• Passion for self-driving technologies.

Benefits

• Work with a world-class team in an innovative environment.
• Opportunity to impact the future of self-driving technology.
• Collaborative and diverse workplace culture.
• Access to cutting-edge technology and tools.
• Flexible working arrangements including remote work options.