Staff Machine Learning Engineer, Generative AI Modeling, Inference

• Develop innovative machine learning technology and products serving 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

• Proven passion for machine learning and staying current with research
• Familiarity with neural networks, deep learning, and generative modeling
• Deep understanding of mathematics and/or machine learning algorithms
• Ability to solve open, ambiguous problems
• Ability to collaborate effectively with internal teams and external partners
• Ability to work independently
• Bachelor's degree in a technical field such as computer science, mathematics, or 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 plus 7+ years of post-graduate ML or related experience; or a PhD in a related technical field plus 4+ years of post-graduate ML or related experience
• Experience with computer vision or generative modeling techniques
• Experience with TensorFlow, PyTorch, JAX, MLX, scikit-learn, or related frameworks
• Preferred: advanced degree in computer science or related field
• Preferred: experience training large-scale diffusion models for images, videos, or 3D
• Preferred: knowledge of distillation, quantization, and model compression techniques
• Preferred: knowledge of GPU, CPU, or NPU optimization techniques
• Preferred: experience building and optimizing ML inference pipelines

Core Competencies

Demonstrates expertise in machine learning, particularly in generative modeling and deep learning, with a strong foundation in mathematics and algorithms. Proven ability to collaborate across teams and deliver innovative augmented reality experiences.

Highest-signal resume keywords

• Machine Learning Expertise
• Generative Modeling
• Deep Learning
• TensorFlow
• Computer Vision

ATS Optimization Keywords

Hard Skills

• Machine Learning Algorithms
• Neural Networks
• Generative Modeling Techniques
• Mathematics
• Large-Scale Diffusion Models
• Model Compression Techniques
• ML Inference Pipelines
• GPU Optimization
• CPU Optimization
• NPU Optimization

Soft Skills

• Problem Solving
• Collaboration
• Independence

Industry Keywords

• Augmented Reality
• Generative Models
• Cross-Functional Collaboration
• Research in Machine Learning

Tools & Technologies

• TensorFlow
• PyTorch
• JAX
• MLX
• Scikit-learn