Staff Machine Learning Engineer, Generative AI Modeling, Inference

Jobtailor; California (MO); On-site; USD 180,000 - 240,000; Full time

Job summary

Snap Inc. is hiring senior ML engineers to develop generative ML models and on-device AR experiences for millions of Snapchatters.

You will collaborate across cross-functional teams to prototype new products and stay at the forefront of research in neural networks and diffusion techniques.

The role emphasizes solving open problems, tuning inference pipelines, and advancing large-scale diffusion models for image, video, and 3D content, with practical impact.

Qualifications

• Bachelor's degree in a technical field or equivalent experience.
• 8+ years of post-bachelors ML or related experience; or a PhD with 4+ years.
• Experience with computer vision or generative modeling techniques.
• Familiarity with neural networks, deep learning, and generative modeling.
• Ability to solve open, ambiguous problems.
• Ability to collaborate effectively with internal teams and external partners.
• Ability to work independently.
• Experience with TensorFlow, PyTorch, JAX, MLX, scikit-learn, or related frameworks.

Responsibilities

• 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.

Skills

Machine Learning

Deep Learning

Generative Modeling

Computer Vision

TensorFlow

PyTorch

JAX

Scikit-learn

Mathematics

Collaboration

Education

Bachelor's degree in CS/Math/Statistics or equivalent

Master's degree in a technical field

PhD in a related field

Tools

TensorFlow

PyTorch

JAX

MLX

Scikit-learn

Job description

• 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