ML Training Optimization Engineer at Waabi

Enhance autonomous transportation at Waabi as a Senior ML Training Optimization Engineer. Focus on building distributed training frameworks and optimizing model performance for commercial autonomous vehicles.

You will be responsible for creating and maintaining advanced distributed training frameworks at Waabi. With a focus on stability and efficiency, you will profile model runtime and memory, identifying performance bottlenecks. Collaboration with ML engineers and researchers will ensure optimal resource usage while adopting cutting-edge technologies in training and inference frameworks.

Key Responsibilities: • Build and maintain distributed training frameworks for efficient models • Profile runtime and memory to mitigate performance issues • Evaluate new technologies for training and inference integration • Design CUDA kernels and improve deployment techniques • Collaborate on best practices for resource optimization

Requirements: • MS/PhD or Bachelor’s de...