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 production
Profile model runtime and memory to identify bottlenecks
Evaluate CUDA kernels, quantization, and deployment techniques
Develop training and inference framework improvements
Create tooling and dashboards supporting broad adoption
Requirements: MS, PhD, or bachelor’s degree in a relevant technical field
At least four years of industry experience with a bachelor’s degree
Proficiency in Python, C++, or Rust
Experience with PyTorch or Jax
CPU and GPU profiling experience using stated tools
Apply distributed training, profiling, and deep learning expertise to advance Waabi’s research and production systems.

#J-18808-Ljbffr

📌 Senior Distributed Training Engineer (Ontario)
🏢 Waabi
📍 Ontario