Build distributed training frameworks at Waabi, applying Python, C++, or Rust to improve machine learning stability and efficiency.As a Senior or Staff Machine Learning Training Optimization Engineer, you will support research and production training. You will profile model runtime and memory, evaluate emerging technologies, and collaborate with researchers and machine learning engineers. Your work will advance training and inference frameworks for autonomous transportation.Key Responsibilities:
Build standardized distributed training frameworks for research and productionProfile model runtime and memory to identify performance bottlenecksEvaluate CUDA kernels, quantization, and compilation techniquesDevelop tooling and dashboards supporting broad adoptionEstablish resource usage best practices with technical teamsRequirements:
MS, PhD, or bachelor's degree in a technical fieldAt least four years of industry experience with bachelor's degreeProficiency in Python, C++, or RustExperience with PyTorch or JaxCPU and GPU profiling experience using relevant toolsApply training optimization, distributed systems, and profiling expertise to Waabi’s autonomous transportation technology.
J-18808-Ljbffr