Staff ML Engineer: Diffusion & Generative Modeling

Relha LLC; Los Angeles, Northern (CA, KY); Hybrid; USD 195,000 - 343,000; Full time

Benefits offered by this job

Equity (RSUs)

Job summary

Snap Inc. is seeking a Staff Machine Learning Engineer to advance diffusion, generative modeling, and on-device inference across the Generative ML Platform. You will develop ML technology powering Snapchat experiences and collaborate with cross-functional teams to prototype new products.

The role requires 8+ years of ML experience, strong background in computer vision and generative models, and deep knowledge of TensorFlow, PyTorch and JAX. Equity is available, with potential relocation support.

Qualifications

• Bachelor's degree in computer science, math, statistics or equivalent.
• 8+ years of post-Bachelor’s ML or related experience; or MS +7+ years; or PhD +4+ years.
• Experience with Computer Vision or Generative Modeling techniques.
• Experience with TensorFlow, PyTorch, JAX, MLX, scikit-learn or related frameworks.

Responsibilities

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

Skills

Machine Learning

Deep Learning

Generative Modeling

Computer Vision

Python

Education

Bachelor's degree in technical field

Master's degree in a technical field

PhD in a related technical field

Tools

TensorFlow

PyTorch

JAX

scikit-learn

ML frameworks

Job description

Snap Inc. is seeking a Staff Machine Learning Engineer to advance diffusion, generative modeling, and on-device inference across the Generative ML Platform. You will develop ML technology powering Snapchat experiences and collaborate with cross-functional teams to prototype new products.

The role requires 8+ years of ML experience, strong background in computer vision and generative models, and deep knowledge of TensorFlow, PyTorch and JAX. Equity is available, with potential relocation support.