ML Infrastructure Engineer: Scalable AI Platform

Socket.dev; Santa Monica (CA); On-site; USD 157,000 - 235,000; Full time

Job summary

Snap Inc. is hiring a Software Engineer for the ML Platform Experience team within the core ML Platform org. You will design and optimize infrastructure for ML workloads at scale, build feature pipelines for online and offline data, and develop high-performance inference and training systems in the cloud.

You’ll work with Python and Java, collaborate with ML engineers to deploy models to production, and help maintain scalable, secure, and reliable systems across Snap’s AI tooling stack.

Qualifications

• Bachelor's degree in CS or equivalent required.
• 2+ years of post-Bachelor's software development experience.
• Experience building large-scale production ML systems and distributed/big data processing.

Responsibilities

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

Skills

Python

Java

Distributed systems

Big data frameworks

Cloud infrastructure

Collaboration

Production systems

Education

Bachelor's degree in Computer Science or equivalent

Master's degree or PhD preferred

Tools

Spark

Flink

Ray

Job description

Snap Inc. is hiring a Software Engineer for the ML Platform Experience team within the core ML Platform org. You will design and optimize infrastructure for ML workloads at scale, build feature pipelines for online and offline data, and develop high-performance inference and training systems in the cloud.

You’ll work with Python and Java, collaborate with ML engineers to deploy models to production, and help maintain scalable, secure, and reliable systems across Snap’s AI tooling stack.