Staff Machine Learning Engineer - Dataset & Training Platform - Remote

Overview

Join the team redefining how the world experiences design.

In Short

• Architecting foundational AI Platform capabilities.
• Leading cross-team initiatives for dataset and training improvements.
• Building high-performance distributed data processing systems.
• Driving technical strategy discussions for platform stability.
• Mentoring engineers and enhancing AI platform capabilities.
• Solving complex challenges in the ML stack.

Requirements

• 5+ years in ML training systems and data processing.
• Experience in AI/ML infrastructure design and implementation.
• Proven track record in setting technical direction.
• Strong understanding of distributed computing and cloud infrastructure.
• Fluent in Python with knowledge of ML frameworks.
• Understanding of generative AI systems.
• Ability to balance platform investments with business needs.
• Experience with product teams and diverse technical requirements.
• Track record of growing engineers and contributing to culture.
• Experience with infrastructure-as-code and performance optimization.

Benefits

• Equity packages for shared success.
• Inclusive parental leave policy.
• Annual Vibe & Thrive allowance for wellbeing and office setup.
• Flexible leave options for personal support.