ML Infrastructure Engineer — Scale Fast AI Platforms

Snap Inc.; Palo Alto (CA); On-site; USD 133,000 - 235,000; Full time

Benefits offered by this job

Paid parental leave

Comprehensive medical coverage

Mental health support programs

Equity in RSUs

Job summary

Snap Inc. is seeking a Software Engineer to join the ML Platform Experience team, part of the core ML Platform organization. You will design and optimize ML infrastructure, build pipelines for online inference and offline data, and collaborate with ML engineers to deploy models.

The role emphasizes Python and Java, scalable systems, and high-performance inference. The company offers equity in RSUs and a generous compensation package, with a standard in-office schedule in California.

Qualifications

• Bachelor’s degree in computer science or equivalent experience.
• 2 years of post-Bachelor’s software development experience, or 1 year with Master’s/PhD.
• Experience building large-scale production ML systems and distributed systems.
• Strong Python/Java programming for ML platforms.
• Ability to learn new AI concepts and apply them.

Responsibilities

• Design and optimize infrastructure for ML workloads at scale.
• Build feature generation and serving pipelines for online inference and offline data.
• Develop high-performance inference systems for AI models.
• Enable scalable ML model training, evaluation, and inference in the cloud.
• Create data management systems for data collection, labeling, processing, and evaluation.
• Collaborate with ML engineers to deploy models into production.
• Uphold code quality, security, and production-readiness standards.

Skills

Python

Java

Distributed systems

Big data processing

Spark

Flink

Ray

Collaboration

Education

Bachelor’s degree in computer science or equivalent

Master’s degree in a technical field

PhD in a relevant technical field

Tools

TensorFlow

PyTorch

Caffe2

Spark ML

scikit-learn

Job description

Snap Inc. is seeking a Software Engineer to join the ML Platform Experience team, part of the core ML Platform organization. You will design and optimize ML infrastructure, build pipelines for online inference and offline data, and collaborate with ML engineers to deploy models.

The role emphasizes Python and Java, scalable systems, and high-performance inference. The company offers equity in RSUs and a generous compensation package, with a standard in-office schedule in California.