Software Engineer, ML Infrastructure

Snap Inc.

Software Engineer optimizing machine learning infrastructure for Snap Inc. in the United States, collaborating on large-scale systems for AI model serving and training.

Posted 5/29/2026full-timeCalifornia, Washington • 🇺🇸 United StatesJuniorMid-Level💰 $157,000 - $235,000 per yearWebsite

Tech Stack

Tools & technologies

CloudDistributed SystemsJavaPythonRayScalaSpark

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
Build and enhance feature generation and serving pipelines that power online inferencing and offline training data generation
Develop high-performance inference systems to ensure fast and efficient AI model serving
Build infrastructure to perform scalable ML model training, evaluation, and inference in the cloud
Develop high-performance inference systems to ensure fast and efficient AI model serving
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
2+ years of post-Bachelor’s software development experience; or Master’s degree in a technical field + 1+ year of post-grad software development experience; or PhD in a relevant technical field
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
Experience with big data processing frameworks such as Spark, Flink, or Ray

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 learningdistributed systemsbig data processingSparkFlinkRay

Soft Skills

problem-solvingfocus on system performancescalabilityefficiency

Certifications

Bachelor's degreeMaster's degreePhD