Staff Machine Learning Engineer, ML Efficiency

📋 Description

• Design and build systems to improve ML training and inference efficiency.
• Develop tooling to debug, profile, optimize, and monitor model performance.
• Improve GPU and resource utilization via scheduling, caching, and workload optimization.
• Partner with ML researchers and product teams to identify bottlenecks and boost performance.
• Build benchmarking frameworks and dashboards for training and serving systems.
• Optimize distributed training infrastructure and model serving architectures.

🎯 Requirements

• BS, MS, or PhD in Computer Science or related field.
• 5+ years of software engineering experience.
• Strong proficiency in Python.
• Proficiency in at least one systems language (Go, C++, Rust, or Java).
• Experience building distributed systems at scale.
• Experience with ML infrastructure, training systems, or model serving platforms.

🎁 Benefits

• Global Benefit programs that fit your lifestyle, from workspace to professional development to caregiving support
• Family Planning Support
• Mental Health & Coaching Benefits
• Group Personal Pension Scheme with Employer match
• Private Medical and Dental Scheme
• Flexible Vacation & Paid Volunteer Time Off