Principal Machine Learning Engineer, ML Training Platform

Job expired!

Join Snap Inc. as a Principal Machine Learning Engineer, ML Training Platform

Snap Inc. is a leading technology company revolutionizing the way people live and communicate. We believe the camera is a powerful tool to enhance human expression and connection. Our mission is to contribute to human progress by empowering people to express themselves, live in the moment, learn about the world, and have fun together. Our core products include:

  • Snapchat: A visual messaging app that strengthens relationships with friends, family, and the global community.
  • Lens Studio: An augmented reality platform powering AR experiences across Snapchat and other services.
  • Spectacles: Innovative AR glasses that bring augmented reality to life.

About the Role

We are looking for a Principal Machine Learning Engineer to join our team. In this role, you will design, implement, and scale critical machine learning components and services to support Snap's most strategic initiatives. Your work will push the limits of what's possible with machine learning.

Responsibilities

  • Build a next-generation training framework to support large-scale model training.
  • Optimize training and model performance with various GPUs for increased speed and efficiency.
  • Develop an AutoML platform to streamline the machine learning model lifecycle.
  • Collaborate with cross-functional teams to understand product requirements and deliver innovative solutions.
  • Advocate for best practices in availability, scalability, operational excellence, and cost management.
  • Provide technical direction that impacts the entire company.

Knowledge, Skills & Abilities

  • Strong understanding of machine learning approaches and algorithms.
  • Excellent programming and software design skills (debugging, performance analysis, and test design).
  • Proven track record of operating highly-available systems at scale.
  • Ability to quickly learn new concepts and technologies.
  • Strong problem-solving and collaboration skills.

Minimum Qualifications

  • BS in a technical field (computer science, mathematics, statistics) or equivalent experience.
  • 14+ years of industry machine learning experience.
  • Experience with GPU/TPU training and optimizations.

Preferred Qualifications

  • Master’s/PhD in a technical field (computer science).
  • Experience leading teams and driving technical roadmaps.
  • Experience with machine learning, recommendation systems, or vector similarity search.
  • Proficiency with TensorFlow, PyTorch, or related deep learning frameworks.
  • Experience with Docker, Kubernetes, Ray, NoSQL solutions, Memcache/Redis, Google/AWS services.
  • Experience in MLOps and managing the production machine learning lifecycle.

Inclusion and Diversity

If you have a disability or special need that requires accommodation, please let us know.

At Snap Inc., we believe that being together in person fosters dynamic collaboration and helps us build our culture faster. Our "Default Together" policy encourages team members to work in an office 4+ days per week.

We are proud to be an equal opportunity employer, committed to creating a diverse and inclusive environment. All qualified applicants will receive employment consideration regardless of race, religious creed, color, national origin, ancestry, physical disability, mental disability, medical condition, genetic information, marital status, sex, gender, gender identity, gender expression, pregnancy, age, sexual orientation, military or veteran status, or any other protected classification, in accordance with applicable laws.

Our Benefits

Snap Inc. offers a comprehensive benefits package, including paid parental leave, medical coverage, mental health support programs, and compensation packages that include equity options.

Compensation

Snap Inc. assigns work locations a pay zone, determining the salary range for the position. Starting pay is based on skills, experience, qualifications, work location, and market conditions. Pay zones may be modified in the future.

  • Zone A (CA, WA, NYC): $244,000-$366,000 annually.
  • Zone B: $232,000-$348,000 annually.
  • Zone C: $208,000-$311,