Optimize Waabi’s distributed machine learning training with profiling, CUDA, and inference technologies across research and production.
This Senior or Staff engineering role focuses on creating stable, efficient training frameworks for Waabi’s autonomous transportation technology. You will investigate CPU and GPU performance, assess quantization-aware training, and explore compilation and deployment techniques. Collaboration with researchers and machine learning engineers will guide optimal resource usage and practical adoption through tooling and dashboards.
Key Responsibilities
Drive distributed training toward greater stability and efficiencyComprehensively profile runtime and memory performanceIdentify technologies for training and inference frameworksDesign current CUDA kernels when appropriateImprove dashboards that encourage broad tool adoptionRequirements:
Computer Science, Robotics, or similar technical educationMinimum four years industry experience with bachelor's degreeStrong coding skills across Python, C++, or RustDeep learning framework experience with PyTorch or JaxProfiling experience with PyTorch Profiler or NVIDIA NsightBring distributed training, performance profiling, and framework optimization experience to Waabi’s growing engineering team.
#J-18808-Ljbffr