Staff ML Engineer - Generative AI & On-Device Inference

Snapchat; Santa Clarita (CA); On-site; USD 195,000 - 343,000; Full time

Benefits offered by this job

Paid parental leave

Medical coverage

Mental health support

Equity (RSUs)

Job summary

Snap Inc. seeks an experienced Machine Learning Engineer to build innovative AI-powered features for Snapchat, including on-device generative models and AR experiences. You will collaborate with cross-functional teams to prototype and deliver cutting-edge ML products at scale.

The role emphasizes deep learning, diffusion models, and efficient ML pipelines, with a strong focus on research, prototyping, and professional growth in a collaborative environment.

Qualifications

• Bachelor’s degree in a technical field and 8+ years of ML experience or 7+ years with a Master’s or 4+ with a PhD.
• Experience with Computer Vision or Generative Modeling techniques.
• Experience with ML frameworks such as TensorFlow, PyTorch, JAX, or scikit-learn.

Responsibilities

• Develop innovative machine learning technology and products for millions of Snapchatters.
• Build cutting-edge AR experiences using generative models.
• Deliver generative ML experiences on device and in production environments.
• Collaborate with cross-functional teams to prototype new products.

Skills

Machine Learning

Neural Networks

Deep Learning

Generative Modeling

Mathematics

Collaboration

Independent Work

Education

Bachelor's degree in CS/Math/Stats

Tools

TensorFlow

PyTorch

JAX

ML Frameworks

Diffusion Models

Job description

Snap Inc. seeks an experienced Machine Learning Engineer to build innovative AI-powered features for Snapchat, including on-device generative models and AR experiences. You will collaborate with cross-functional teams to prototype and deliver cutting-edge ML products at scale.

The role emphasizes deep learning, diffusion models, and efficient ML pipelines, with a strong focus on research, prototyping, and professional growth in a collaborative environment.