Senior ML Infra Engineer - High-Performance AI Platform

SNAP, Inc.; Palo Alto (CA); On-site; USD 209,000 - 313,000; Full time

Benefits offered by this job

Paid parental leave

Comprehensive medical coverage

Emotional and mental health support

RSU equity

Job summary

Snap Inc. is seeking a Software Engineer to join the ML Platform team in California. You will scale ML infrastructure, optimize AI training and inference, and drive innovations for efficient ranking and recommendations across Snapchat.

The role focuses on building data platforms, feature management, and scalable data processing. Candidates should have strong programming skills and a track record of operating large-scale, highly available systems.

Qualifications

• Bachelor’s degree in a technical field or equivalent experience.
• 6+ years of post-Bachelor’s software development experience; or 5+ with a Master’s; or PhD + 2+ years.
• Experience building large-scale production ML systems, distributed systems or big data processing.

Responsibilities

• Design and optimize infrastructure systems for ML workloads at scale and improve reliability and efficiency across Snapchat’s ML Infrastructure
• Develop high-performance inference systems to ensure fast and efficient AI model serving
• Build infrastructure for scalable ML model training, evaluation, and inference in the cloud
• Develop data management systems for scalable data collection, labeling, processing, and evaluation
• Work on vector search algorithms to improve precision, recall and scalability of retrieval systems
• Collaborate with ML engineers to deploy cutting-edge models into production

Skills

Python

Java

Scala

C++

Distributed systems

System performance

Collaboration

Problem-solving

Education

Bachelor’s degree in computer science or equivalent

Advanced degree (Masters/PhD) in technical field

Tools

Spark

Flink

Ray

TensorFlow

PyTorch

Caffe2

Spark ML

scikit-learn

Job description

Snap Inc. is seeking a Software Engineer to join the ML Platform team in California. You will scale ML infrastructure, optimize AI training and inference, and drive innovations for efficient ranking and recommendations across Snapchat.

The role focuses on building data platforms, feature management, and scalable data processing. Candidates should have strong programming skills and a track record of operating large-scale, highly available systems.