Lead Machine Learning Engineer, Generative AI (Image & Video)

Overview

In this role you will lead ML-powered consumer experiences, shaping GenAI-enabled products and AI Lenses for Snapchat. You’ll set the technical roadmap, own architecture for diffusion-based pipelines, and drive projects from prototype to production. You will collaborate with product, software, Lens Content, and data science to improve experiences through testing and user feedback. This role is a hands-on leadership position focused on delivering scalable, high-quality visual AI features that engage millions of Snapchatters.

Pay / Benefits

• paid parental leave
• comprehensive medical coverage
• emotional and mental health support programs
• long-term compensation packages

Responsibilities

• Define and steer the technical roadmap and architecture for GenAI pipelines (diffusion and flow matching)
• Lead complex ML projects from prototype to production while staying hands-on
• Collaborate with Product, Software Engineering, Lens Content, and Data Science to optimize experiences via A/B testing and feedback
• Evaluate open-source models and third-party APIs; guide build-versus-buy decisions
• Mentor engineers through design reviews, collaboration, and code reviews; build reusable capabilities for faster delivery

Key requirements

• Strong background in ML with experience in generative models, CV systems, or related deep learning applications
• Proficiency in Python or C++, with PyTorch or similar frameworks
• Bachelor’s degree in CS, mathematics, statistics, or related field, or equivalent practical experience
• 5+ years of ML experience post-bachelor (or equivalent) and track record of delivering ML projects to production
• Ability to set technical direction and lead initiatives through ambiguity

• Excellent written and verbal communication; ability to explain trade-offs and align stakeholders
• Strong product judgment and attention to visual quality
• Mentorship and collaboration across cross-functional teams
• Generative modeling
• Computer vision systems
• PyTorch (or comparable DL framework)

…