Staff Software Engineer, ML Infrastructure, Level 6

About the role

Snap Inc. is seeking a Staff Software Engineer, ML Infrastructure to join our team. This role plays a critical part in scaling our ML Infrastructure, optimizing our embedding, feature and training data storage and compute to enable different paradigms of ML models at massive scale, and improving Snapchat's ranking and recommendation systems.

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 embedding generation / batch inference systems to improve model performance
• Develop high-performance data storage/compute systems to improve the efficiency of our ML infrastructure
• Integrate state of the art ML data quality system to assure model performance
• Build comprehensive data management systems for scalable data collection, labeling, processing, and evaluation
• Work closely with ML engineers to deploy cutting-edge models into production
• Utilize AI tools and high velocity engineering workflows to design and ship scalable services while upholding rigorous standards for code correctness, security, and production ready quality code

Requirements

• 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
• 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
• Proficiency in, or a strong aptitude for, leveraging AI tools to streamline development, paired with the critical judgment to audit generated output for architectural integrity, performance bottlenecks, and security risks
• Adaptability in learning and applying evolving AI systems and tools to remain at the forefront of engineering trends and modern development practices

Qualifications

• Bachelor’s degree in a technical field such as computer science or equivalent experience
• 9+ years of post-Bachelor’s software development experience; or Master’s degree in a technical field + 5+ years 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 with big data processing frameworks such as Spark, Flink, or Ray
• Familiarity with ML frameworks such as Pytorch, Tensorflow

Benefits

We offer a comprehensive benefits package including paid parental leave, comprehensive medical coverage, emotional and mental health support programs, and compensation packages that let you share in Snap’s long-term success!