Software Engineer, ML Infrastructure

Responsibilities

• Design and optimize infrastructure systems for machine learning workloads at scale to drive reliability and efficiency.
• Build and enhance feature generation and serving pipelines for online inferencing and offline training data generation.
• 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 comprehensive data management systems for scalable data collection, labeling, processing, and evaluation.
• Collaborate closely with ML engineers to deploy cutting-edge models into production.

Requirements

• 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 + 1 year, or PhD).
• Experience building large-scale production machine learning systems, distributed systems, or big data processing.
• Proficiency in Python, Java, Scala, or C++.
• Strong understanding of distributed systems and the infrastructure components of large-scale ML.

Preferred Qualifications

• Master's or PhD in a technical field such as computer science.
• 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 big data processing frameworks such as Spark, Flink, or Ray.

Benefits

• Comprehensive medical coverage.
• Paid parental leave.
• Emotional and mental health support programs.
• Compensation packages including equity in the form of RSUs.

About the Company

Snap Inc. is a technology company that believes the camera presents the greatest opportunity to improve the way people live and communicate. Our core products include Snapchat, Lens Studio, and Spectacles. We are committed to empowering people to express themselves, live in the moment, and have fun together through technically sophisticated products.