Staff Machine Learning Engineer, Generative AI Modeling and Inference

About the role

Snap Inc. is a technology company focused on empowering people to express themselves, live in the moment, learn about the world, and have fun together through innovative products like Snapchat, Spectacles, Bitmoji, and Saturn. The Generative ML Platform team at Snap builds cutting-edge AI technologies, including multimodal LLMs, video generation, real-time AR, human understanding, and 3D content creation. We develop the full stack of generative AI—from foundational models to efficient infrastructure and on-device/server-side inference—to power creative, scalable experiences for hundreds of millions of Snapchatters worldwide.

Responsibilities

• Develop innovative machine learning technology and products that serve millions of Snapchatters
• Build cutting-edge augmented reality experiences using generative models
• Deliver generative machine learning experiences on device
• Partner with cross-functional Snap teams to explore and prototype new products

Requirements

• Bachelor's degree in a technical field such as computer science, mathematics, statistics, or equivalent years of experience
• 8+ years of post-Bachelor’s machine learning or related experience; or a Master’s degree in a technical field + 7+ years of post-grad ML or related experience; or a PhD in a related technical field + 4+ years of post-grad ML or related experience
• Experience with Computer Vision or Generative Modeling techniques
• Experience working with machine learning frameworks such as TensorFlow, PyTorch, JAX, MLX, scikit-learn, or related frameworks

Skills

• A proven passion for machine learning; staying up-to-date with research and prototyping new ideas quickly
• Familiarity with Neural Networks, Deep Learning, and Generative Modeling
• Deep understanding of mathematics and/or machine learning algorithms
• Desire to solve open, ambiguous problems
• Commitment to professional growth, learning, and supporting others
• Ability to collaborate effectively with internal teams and external partners
• Ability to work independently

Preferred Qualifications

• Advanced degree in computer science or a related field
• Experience training large-scale diffusion models for images, videos, or 3D
• Knowledge of distillation, quantization, and model compression techniques
• Knowledge of GPU, CPU, or NPU optimization techniques
• Experience building and optimizing ML inference pipelines

Benefits

Snap Inc. offers benefits designed to support your well-being and long-term success, including:

• Paid parental leave
• Comprehensive medical coverage
• Emotional and mental health support programs
• Compensation packages that let you share in Snap’s long-term success

Pay

In the United States, work locations are assigned a pay zone which determines the salary range for the position. The successful candidate’s starting pay will be based on job-related skills, experience, qualifications, work location, and market conditions, and may be negotiable within the salary range.

• Zone A (CA, WA, NYC): $229,000–$343,000 annually
• Zone B: $218,000–$326,000 annually
• Zone C: $195,000–$292,000 annually

This position is also eligible for equity in the form of RSUs.

Schedule

Snap Inc. practices a “default together” approach, expecting team members to work in an office 4+ days per week to foster collaboration and culture.