Senior Distributed Training Engineer

Build distributed training frameworks at Waabi using Python, C++, or Rust to improve machine learning stability and efficiency. Work from Waabi’s stated offices.As a Senior ML Training Optimization Engineer, you will support research and production training systems at Waabi. The work combines deep learning frameworks, runtime profiling, memory analysis, and resource optimization. You will collaborate with researchers and ML engineers while improving tools for broad adoption.Key Responsibilities:
Build standardized distributed training frameworks for research and productionProfile model runtime and memory to identify bottlenecksEvaluate CUDA kernels, quantization, and deployment techniquesDevelop training and inference framework improvementsCreate tooling and dashboards supporting broad adoptionRequirements:
MS, PhD, or bachelor’s degree in a relevant technical fieldAt least four years of industry experience with a bachelor’s degreeProficiency in Python, C++, or RustExperience with PyTorch or JaxCPU and GPU profiling experience using stated toolsApply distributed training, profiling, and deep learning expertise to advance Waabi’s research and production systems.
#J-18808-Ljbffr