Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference

Overview In this role you will develop cutting-edge machine learning technologies that power Snapchat experiences. You will build AR features using generative models and deliver on-device ML experiences, collaborating with cross-functional teams to prototype new products. You’ll contribute to a mission-driven platform that enables creators and users to express themselves with AI-enabled visuals. This is an opportunity to work at scale on multimodal models and real-time experiences that shape how people live and communicate. Compensation / Benefitspaid parental leavecomprehensive medical coverageemotional and mental health supportequity (RSUs)salary ranges by zoneinclusive workplace policies ResponsibilitiesDevelop and ship innovative ML-driven products for millions of Snapchat usersCreate AR experiences with generative modelsDeliver on-device generative ML capabilitiesCollaborate with cross-functional teams to explore and prototype new products Key requirementsBachelor's in Computer Science (or equivalent)8+ years post-Bachelor’s ML experience or 7+ years with Master or 4+ years with PhDExperience with Computer Vision or Generative Modeling techniquesProficiency with ML frameworks (TensorFlow, PyTorch, JAX, MLX, scikit-learn)Strong fundamentals in mathematics/ML algorithmsAbility to work independently and collaborate with internal and external partnersDemonstrated passion for ML and ability to prototype quicklyDesire to solve open ambiguous problemscollaborationgrowth mindsetindependent workNeural Networks and Deep LearningGenerative ModelingComputer Vision