Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference

Snap Inc.

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Bellevue, WALos Angeles, CANew York, NYPalo Alto, CASeattle, WA
Salary
$229,000–$343,000 / yr
Posted
27 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $226k
This role $286k
$166k most similar roles pay here $362k

This role pays more than 88% of similar roles. Most pay $197,925–$254,750 — the shaded band above. At the midpoint, this role pays about $286k versus about $226k for comparable roles.

Based on 240 similar postings.

Employer

About Snap Inc.

Snap Inc. is a technology and camera company, best known for Snapchat, offering visual communication, augmented reality, and advertising products.

Snap Inc. currently has 76 open roles on FindRole.

Listed pay typically runs $209,000–$313,000 across 58 roles with salary data.

Most-posted roles

View all roles at Snap Inc.

At a glance

TL;DR · Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference

Staff Machine Learning Engineer, Diffusion, Generative Modeling and Inference joins the Generative ML Platform team to develop cutting-edge AI technologies. This role involves building innovative machine learning products, creating augmented reality experiences using generative models, and delivering generative machine learning capabilities on mobile devices. The engineer will collaborate with cross-functional teams to prototype new products while solving complex, ambiguous problems in the realm of multimodal LLMs, video generation, and 3D content creation. Required expertise includes deep knowledge of neural networks, deep learning, and generative modeling techniques. Candidates should be proficient in machine learning frameworks such as TensorFlow, PyTorch, JAX, MLX, or scikit-learn. Preferred skills include experience with large-scale diffusion models for images and video, model compression via distillation and quantization, and optimizing inference pipelines across GPU, CPU, and NPU hardware architectures.

What does a Machine Learning Engineer earn in Washington?

Median $241750 from 66 postings across 14 companies.

See salary data

What you'll do

  • Develop innovative machine learning technologies and products for millions of Snapchat users.
  • Build cutting-edge augmented reality experiences using generative models.
  • Deliver generative machine learning experiences directly on mobile devices.
  • Prototype new products in collaboration with cross-functional teams.
  • Train large-scale diffusion models for images, videos, or 3D content.
  • Optimize ML inference pipelines and implement model compression techniques like distillation and quantization.
  • Optimize performance across GPU, CPU, and NPU hardware architectures.

What we're looking for

  • Bachelor's degree in a technical field like computer science or mathematics, or equivalent experience.
  • 8+ years of post-Bachelor’s machine learning experience, or a Master’s with 7+ years, or a PhD with 4+ years.
  • Experience with Computer Vision or Generative Modeling techniques.
  • Experience working with machine learning frameworks such as TensorFlow, PyTorch, JAX, MLX, or scikit-learn.
  • 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, model compression, and GPU/CPU/NPU optimization techniques (preferred).
  • Experience building and optimizing ML inference pipelines (preferred).

More like this

Similar roles

Machine Learning Engineer, Generative AI

Qualcomm

San Diego, CA 39 days ago $104,000$156,000
Generative AI LLM RAG Python PyTorch TensorFlow LangChain LlamaIndex Transformers Vector Databases ML ops Microservices AWS GCP Azure Reinforcement Learning Prompt Engineering Agile

Staff Machine Learning Engineer

Intuit

Mountain View, CA 152 days ago $202,500$274,000
Python TensorFlow Keras Spark SQL NLTK NumPy Pandas AWS GCP Git GitHub Machine Learning Operations Data Pipelines R Data Structures Algorithms
7+ yrs exp

Staff Machine Learning Engineer

Intuit

Mountain View, CA 147 days ago $202,500$274,000
Python NLTK NumPy Pandas TensorFlow Keras R Spark SQL Git GitHub CUDA cuDNN AWS GCP Machine Learning Feature Engineering
6+ yrs exp

Staff Machine Learning Engineer

GEICO

Palo Alto, CA +1 98 days ago $130,000$260,000
Generative AI LLMs Python Java C++ C# Kubernetes CICD Kafka Spark Ray Airflow Temporal PostgreSQL MongoDB Cassandra Snowflake elastic search Qdrant
6+ yrs exp

Staff Machine Learning Engineer

Adobe

San Jose, CA 77 days ago $211,800$306,625
Python PyTorch TensorFlow JAX Hugging Face Ray Spark LLMs multimodal foundation models Vector Search Information Retrieval MLOps Distributed Training A/B Testing Transformer Architectures Representation Learning Metric Learning search, ranking

Staff Machine Learning Engineer

Arm Holdings

Austin, TX 32 days ago $249,900$338,100
Machine Learning Python TensorFlow PyTorch ARM Architecture GPU Neural Networks Computer Architecture hardware aware optimization Deep Learning
Hybrid