Software Engineer, ML Infrastructure, Level 5

Position Overview

You'll play a critical role in scaling our ML Infrastructure, optimizing AI training and inference systems, and driving innovations that make Snapchat's ranking and recommendation systems more efficient and impactful. We're looking for a Software Engineer, ML Infrastructure to join Snap Inc!

Responsibilities

• Design and optimize infrastructure systems for machine learning workloads at scale and drive reliability and efficiency improvements across Snapchat's ML Infrastructure
• 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 on state-of-the-art vector search algorithms to improve the precision, recall and scalability of our retrieval systems
• Work closely with ML engineers to deploy cutting-edge models into production

Requirements & Skills

Knowledge, Skills & Abilities:

• Strong programming skills in Python, Java, Scala or C++
• 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

Minimum Qualifications:

• Bachelor's degree in a technical field such as computer science or equivalent experience
• 6+ years of post-Bachelor's software development experience; or Master's degree in a technical field + 5+ year of post-grad software development experience; or PhD in a relevant technical field+ 2+ years of post-grad software development experience
• 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