Staff Software Engineer, ML Infrastructure

SNAP/SNAP

Staff ML Infrastructure Engineer scaling production machine-learning systems for Snapchat at Snap. Designing embedding, feature-store, data-storage, and inference infrastructure at massive scale.

Posted 8/11/2026full-timePalo Alto • California, Washington • 🇺🇸 United StatesLead💰 $229,000 - $343,000 per yearWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in designing and optimizing infrastructure systems for machine learning workloads, with a strong focus on reliability, efficiency, and scalability. Proficient in leveraging AI tools and developing high-performance data management systems to support large-scale ML applications.

Highest-signal resume keywords

Python ProgrammingMachine Learning InfrastructureDistributed SystemsAI Tools ProficiencyBachelor's Degree in Technical Field

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

Java ProgrammingScala ProgrammingC++ ProgrammingSystem Performance OptimizationBatch Inference Systems DevelopmentData Storage Systems DevelopmentData Management SystemsLarge-Scale Production SystemsBig Data ProcessingFeature Store Development

Soft Skills

Problem-SolvingCollaborationAdaptability

Tools & Technologies

SparkFlinkRayPyTorchTensorFlow

Industry Keywords

Machine LearningInfrastructure SystemsCode CorrectnessSecurity StandardsModel Performance Assurance

Tech Stack

Tools & technologies

Distributed SystemsJavaPythonPyTorchRayScalaSparkTensorflow

About the role

Key responsibilities & impact

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

Requirements

What you’ll need

Strong programming skills in Python, Java, Scala, or C++
Strong problem-solving skills focused on system performance, scalability, and efficiency
Good understanding of distributed systems and large-scale ML infrastructure
Ability to collaborate and work well with others
Proven track record operating highly available systems at significant scale
Ability to proactively learn and apply new concepts
Proficiency in, or strong aptitude for, leveraging AI tools and auditing generated output for architecture, performance, and security
Adaptability in learning and applying evolving AI systems and tools
Bachelor’s degree in a technical field such as computer science or equivalent experience, plus 9+ years of post-Bachelor’s software development experience; or Master’s degree in a technical field plus 5+ years of post-graduate software development experience; or PhD in a relevant technical field plus 2+ years of post-graduate software development experience
Experience building large-scale production machine learning systems, distributed systems, or big data processing
Preferred: Master’s/PhD in a technical field or equivalent industry experience
Preferred: Experience with Spark, Flink, or Ray
Preferred: Experience with large-scale feature stores or embedding systems
Preferred: Familiarity with PyTorch or TensorFlow

Benefits

Comp & perks

Paid parental leave
Comprehensive medical coverage
Emotional and mental health support programs
Compensation packages that let employees share in Snap’s long-term success
Equity in the form of RSUs
Equal opportunity employment