Staff ML Engineer — Generative AI & On-Device AR

Snap; New York (NY); On-site; USD 229,000 - 343,000; Full time

Job summary

Snap Inc. is seeking a Machine Learning Engineer to develop advanced ML technologies that power AR and visual experiences for hundreds of millions of Snapchatters.

You will build on-device generative models and collaborate with cross-functional teams to prototype new products. Strong background in neural networks, deep learning, and generative modeling is required, along with experience in TensorFlow, PyTorch, or related frameworks.

Qualifications

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

Responsibilities

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

Neural networks

Deep learning

Generative modeling

Mathematics

Collaboration

Independent work

Education

Bachelor's degree

Master's degree

PhD

Tools

TensorFlow

PyTorch

JAX

MLX

scikit-learn

Job description

Snap Inc. is seeking a Machine Learning Engineer to develop advanced ML technologies that power AR and visual experiences for hundreds of millions of Snapchatters.

You will build on-device generative models and collaborate with cross-functional teams to prototype new products. Strong background in neural networks, deep learning, and generative modeling is required, along with experience in TensorFlow, PyTorch, or related frameworks.