Senior Engineer Ml Training Optimization Toronto

Transform distributed training as a Senior Engineer focused on ML Optimization at Waabi. Bring innovation in performance metrics and coding expertise, particularly in Python and C++. Waabi is seeking a Senior Engineer to enhance their ML Training Optimization capabilities. The perfect candidate will have an advanced degree and at least four years of industry experience, ideally in Robotics.

Key Responsibilities

• Build and standardize distributed training frameworks for enhanced stability
• Analyze model runtime to identify memory consumption improvements
• Integrate recent technologies into training setups
• Collaborate on resource management best practices across teams
• Innovate tooling and dashboards for better adoption

Requirements

• Degree in Computer Science or related field with 4+ years of experience
• Solid coding skills in Python, C++, or Rust
• Experience with frameworks like PyTorc...