Software Engineer, ML Infrastructure, Level 4

Posted 2 Hours Ago
3 Locations
Hybrid
133K-235K Annually
Junior
Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Snap is a technology company.
The Role
Design, build, and optimize scalable ML infrastructure for training, evaluation, and high-performance inference. Develop feature generation/serving pipelines, data management and labeling systems, and collaborate with ML engineers to deploy production models while ensuring reliability, scalability, and production-quality code.
Summary Generated by Built In

Snap Inc is a technology company. We believe the camera presents the greatest opportunity to improve the way people live and communicate. Snap contributes to human progress by empowering people to express themselves, live in the moment, learn about the world, and have fun together.


The Company operates Snapchat, a visual messaging app that enhances your relationships with friends, family, and the world, and Specs Inc., a wholly-owned subsidiary dedicated to making computing more human, in addition to Bitmoji, Saturn, and other digital services.


Snap Engineering teams build fun and technically sophisticated products that reach hundreds of millions of Snapchatters around the world, every day. We’re deeply committed to the well-being of everyone in our global community, which is why our values are at the root of everything we do. We move fast, with precision, and always execute with privacy at the forefront.

We’re looking for a Software Engineer to join the ML Platform Experience team, part of the core ML Platform organization. We are an AI native team which builds the agentic user experience for building, managing, and operating foundational models at Snapchat. We utilize Python as the primary language with Java/Go as supporting languages, we build agents on ADK, langfuse, and frontier LLM models, and are responsible for many other foundational technology such as model lineage, model orchestration, model data quality, and more.

What you’ll do:

  • 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

Knowledge, Skills & Abilities:

  • Strong programming skills in Python, Java

  • 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

  • Ability to collaborate and work well with others

  • Proven track record of operating highly-available systems at significant scale

  • Ability to proactively learn new concepts and apply them at work

  • Adaptability in learning and applying evolving AI systems and tools to remain at the forefront of engineering trends and modern development practices

Minimum Qualifications:

  • 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

Preferred Qualifications:

  • Masters/PhD in a technical field such as computer science or equivalent industry experience

  • Experience working with ML Training platforms or optimizing AI model inference

  • Familiarity with ML frameworks such as TensorFlow, PyTorch, Caffe2, Spark ML, scikit-learn, or related frameworks

If you have a disability or special need that requires accommodation, please don’t be shy and provide us some information.

"Default Together" Policy at Snap: At Snap Inc. we believe that being together in person helps us build our culture faster, reinforce our values, and serve our community, customers and partners better through dynamic collaboration. To reflect this, we practice a “default together” approach and expect our team members to work in an office 4+ days per week. 

At Snap, we believe that having a team of diverse backgrounds and voices working together will enable us to create innovative products that improve the way people live and communicate. Snap is proud to be an equal opportunity employer, and committed to providing employment opportunities 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, childbirth and breastfeeding, age, sexual orientation, military or veteran status, or any other protected classification, in accordance with applicable federal, state, and local laws. EOE, including disability/vets.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance and the Los Angeles Fair Chance Initiative for Hiring, where applicable).

Our Benefits: Snap Inc. is its own community, so we’ve got your back! We do our best to make sure you and your loved ones have everything you need to be happy and healthy, on your own terms. Our benefits are built around your needs and include paid parental leave, comprehensive medical coverage, emotional and mental health support programs, and compensation packages that let you share in Snap’s long-term success!

Compensation

In the United States, work locations are assigned a pay zone which determines the salary range for the position. The successful candidate’s starting pay will be determined based on job-related skills, experience, qualifications, work location, and market conditions. The starting pay may be negotiable within the salary range for the position. These pay zones may be modified in the future.

Zone A (CA, WA, NYC):

The base salary range for this position is $157,000-$235,000 annually.


 

Zone B:

The base salary range for this position is $149,000-$223,000 annually.

Zone C:

The base salary range for this position is $133,000-$200,000 annually.

This position is eligible for equity in the form of RSUs.

Skills Required

  • Bachelor's degree in a technical field or equivalent experience
  • 2+ years of post-Bachelor's software development experience (or Master's +1/PhD)
  • Experience building large scale production machine learning systems, distributed systems, or big data processing
  • Strong programming skills in Python
  • Strong programming skills in Java
  • Good understanding of distributed systems and infrastructure components of large-scale ML
  • Experience with big data processing frameworks such as Spark, Flink, or Ray
  • Proven track record of operating highly-available systems at significant scale
  • Ability to collaborate and work well with others
  • Ability to proactively learn new concepts and apply them at work
  • Masters or PhD in a technical field
  • Experience working with ML training platforms or optimizing AI model inference
  • Familiarity with ML frameworks such as TensorFlow, PyTorch, Caffe2, Spark ML, or scikit-learn
  • Experience with ADK, Langfuse, or frontier LLM model tooling

What the Team is Saying

Xiaolin
Yvette
Matt
Jasmeet
Xueyin (Sherry)
Amir
Jung
Xu
Talia Mason
Maureen Ufomadu
Vincent Pagnard-Jourdan
Pulkit Trivedi

Snap Inc. Compensation & Benefits Highlights

  • Parental & Family Support Parental leave offers up to 28 weeks with a return-to-work program, alongside adoption, surrogacy, infertility, and fertility-preservation support via partners like Carrot. Additional resources such as backup child care, caregiver assistance (Wellthy), lactation rooms, and breast-milk shipping further strengthen family support.
  • Healthcare Strength Comprehensive medical, dental (including orthodontia), and vision (including LASIK and a frame/contact allowance) coverage is paired with free One Medical membership, virtual physical therapy, and well-being reimbursements. Mental-health access includes 15 annual Lyra sessions for employees and dependents.
  • Retirement Support A 401(k) with immediate vesting and a match of 100% up to 3% plus 50% on the 4th and 5th percent is complemented by pre-tax, Roth, and after-tax contributions including a Mega Backdoor Roth option. HSAs/FSAs, financial education, Rocket Lawyer access, and employee equity round out the financial benefits.

Snap Inc. Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Monica, CA
5,000 Employees
Year Founded: 2011

What We Do

We contribute to human progress by empowering people to express themselves, live in the moment, learn about the world, and have fun together.

Gallery

Gallery

Snap Inc. Teams

Team
Product + Tech
Team
Machine Learning
Team
Sales
About our Teams

Snap Inc. Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Our “default together” approach is an 80/20 model where we are asking team members to spend 80% of the time, on average, in the office, with the remaining 20% of the time spent remote.

Typical time on-site: 4 days a week
HQSanta Monica, CA
Amsterdam, NL
Austin, TX
Bellevue, WA
Berlin, DE
Boulder, CO
Chandler, AZ
Chicago, IL
Copenhagen, DK
Dallas, TX
Eindhoven, NL
Hamburg, DE
London, GB
Mumbai, IN
New York, NY
Oslo, NO
Palo Alto, CA
Paris, FR
San Francisco, CA
Seattle, WA
Singapore
Sydney, AU
Toronto, ON
Vancouver, CA
Washington, DC
Learn more

Similar Jobs

Snap Inc. Logo Snap Inc.

Software Engineer

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
6 Locations
5000 Employees
100K-176K Annually

Snap Inc. Logo Snap Inc.

Software Engineer

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Remote or Hybrid
6 Locations
5000 Employees
133K-235K Annually

Snap Inc. Logo Snap Inc.

Product Manager

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
5 Locations
5000 Employees
178K-313K Annually

Snap Inc. Logo Snap Inc.

Employee Relations Business Partner

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
3 Locations
5000 Employees
111K-196K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account