Senior Engineer in Machine Learning Optimization

Shape technology's future at Waabi as a Senior Engineer in ML Training Optimization. Your role will focus on enhancing distributed training and key performance metrics.
As a Senior Engineer at Waabi, you'll leverage your advanced degree and four years of relevant experience in Robotics to drive innovations in ML Training Optimization. This position will have you build robust frameworks and explore advanced training technologies, ensuring high performance and stability across models.
Key Responsibilities:
• Build and maintain distributed training frameworks for stability
• Analyze runtime to enhance model memory consumption
• Implement recent technologies into training configurations
• Collaborate with teams for optimal resource management
• Create tools and dashboards to improve user adoption
Requirements:
• Degree in Computer Science or applicable field with 4+ years experience
• Strong coding proficiency in Python, C++, or Rust
• Familiarity with PyTo...