Staff Software Engineer, ML Infrastructure

Snap Inc.

Staff Software Engineer optimizing ML Infrastructure and embedding generation systems at Snap Inc. Enhancing model performance and developing data storage solutions for efficient ML operations.

Posted 6/24/2026full-timeBellevue • California, Washington • 🇺🇸 United StatesLead💰 $229,000 - $343,000 per yearWebsite

Tech Stack

Tools & technologies

Distributed SystemsJavaPythonScala

About the role

Key responsibilities & impact

Design and optimize infrastructure systems for machine learning workloads at scale and drive reliability and efficiency improvements across Snapchat’s ML Infrastructure
Develop high-performance embedding generation / batch inference systems to improve model performance
Develop high-performance data storage/compute systems to improve the efficiency of our ML infrastructure
Integrate state of the art ML data quality system to assure model performance
Build comprehensive data management systems for scalable data collection, labeling, processing, and evaluation
Work closely with ML engineers to deploy cutting-edge models into production
Utilize AI tools and high velocity engineering workflows to design and ship scalable services while upholding rigorous standards for code correctness, security, and production ready quality code

Requirements

What you’ll need

Bachelor’s degree in a technical field such as computer science or equivalent experience
9+ years of post-Bachelor’s software development experience; or Master’s degree in a technical field + 5+ years of post-grad software development experience; or PhD in a relevant technical field+ 2+ years of post-grad software development experience
Experience building large scale production machine learning systems, distributed systems or big data processing
Strong programming skills in Python, Java, Scala, or C++
Strong problem-solving skills with a focus on system performance, scalability, and efficiency
Good understanding of distributed systems and the infrastructure components of large-scale ML
Proven track record of operating highly-available systems at significant scale
Ability to proactively learn new concepts and apply them at work
Proficiency in, or a strong aptitude for, leveraging AI tools to streamline development.

Benefits

Comp & perks

paid parental leave
comprehensive medical coverage
emotional and mental health support programs
compensation packages that let you share in Snap’s long-term success

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PythonJavaScalaC++machine learning systemsdistributed systemsbig data processingdata management systemsembedding generationbatch inference

Soft Skills

problem-solvingfocus on system performancescalabilityefficiencyproactive learning

Certifications

Bachelor’s degreeMaster’s degreePhD