Senior Engineer in ML Training Optimization

Elevate your career as a Senior ML Training Optimization Engineer at Waabi. Focus on building advanced distributed training frameworks while enhancing performance and efficiency.

In this senior position, you will possess an MS/PhD or a Bachelor’s degree paired with a minimum of four years in Computer Science or related fields. You must demonstrate proficiency in multiple programming languages like Python and C++, as well as hands-on experience with deep learning tools like PyTorch. Solution-oriented and collaborative, you'll thrive in a role that pushes the boundaries of self-driving tech.

Key Responsibilities:
• Develop robust distributed training frameworks for both research and production
• Profile model runtime and memory to eliminate bottlenecks
• Investigate and implement emerging technologies into training frameworks
• Collaborate with teams for optimal resource utilization
• Design and enhance tools to promote your work ef...