Jobs

Waabi is seeking a Senior / Staff ML Training Optimization Engineer to join their Autonomy & Algorithms team, focusing on enhancing the stability and efficiency of ML training for autonomous transportation. In this role, you will build distributed training frameworks, profile model performance, and evaluate emerging technologies like CUDA kernels and quantization. The ideal candidate will have a strong background in Computer Science, proficiency in Python, C++, or Rust, and experience with deep learning frameworks like PyTorch or Jax. This position offers the opportunity to work on cutting-edge self-driving technologies and contribute to a world-class team.