Staff Software Engineer, ML Infrastructure

Jobtailor; California (MO); On-site; USD 150,000 - 210,000; Full time

Job summary

Snapchat is seeking an experienced software leader to design and optimize infrastructure for machine learning workloads at scale. You will drive reliability and efficiency, develop high-performance embedding generation and batch inference systems, and improve data storage and processing across the ML pipeline.

You will work closely with ML engineers to deploy cutting-edge models into production, applying AI tools and fast development workflows while maintaining rigorous standards for security

Qualifications

• Bachelor’s degree in a technical field or equivalent experience.
• 9+ years of post-Bachelor’s software development experience; or Master’s degree with 5+ years; or PhD with 2+ years.
• Experience building large-scale production machine learning systems, distributed systems or big data processing.
• Strong programming skills in Python, Java, Scala, or C++.

Responsibilities

• Design and optimize infrastructure systems for ML workloads at scale and drive reliability improvements.
• Develop high-performance embedding generation and batch inference systems to improve model performance.
• Develop high-performance data storage/compute systems to improve ML infrastructure efficiency.
• Integrate state-of-the-art ML data quality systems to assure model performance.
• Build comprehensive data management systems for scalable data collection, labeling, processing and evaluation.
• Collaborate with ML engineers to deploy cutting-edge models into production.
• Use AI tools and fast engineering workflows to design and ship scalable services with robust code quality.

Skills

Python

Java

Scala

C++

Distributed systems

Big data processing

Problem solving

AI tools proficiency

Education

Bachelor's degree in a technical field

Master's degree in a technical field

PhD in a technical field

Job description

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

• 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
• 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
• 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.