Software Engineer, ML Infrastructure, Content Retrieval Platform, Level 4

Snap•Palo Alto, CA

7d•Hybrid

About The Position

Snap Inc. is a technology company focused on improving communication through its camera-centric products like Snapchat, Specs Inc., Bitmoji, and Saturn. The company is seeking a Software Engineer to join the Content Retrieval Platform team within the Content ML organization. This team develops and operates a large-scale retrieval system for content recommendation, collaborates with Applied ML teams on architecture and model development, and utilizes languages like Java/Go and C++ for indexing and serving. They are currently building an agentic repository, focusing on retrieval, L0.5, generative retrieval, and redesigning inference feature sourcing.

Requirements

Strong programming skills in Python, Java
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
Adaptability in learning and applying evolving AI systems and tools
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 degree in a technical field + 1+ year of post-grad software development experience; or PhD in a relevant technical field
Experience building large scale production machine learning systems, distributed systems or big data processing

Nice To Haves

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

Responsibilities

Design and optimize infrastructure systems for machine learning workloads at scale.
Drive reliability and efficiency improvements across Snapchat’s ML Infrastructure.
Build and enhance feature generation and serving pipelines for online inferencing and offline training data generation.
Develop high-performance inference systems for fast and efficient AI model serving.
Build infrastructure for scalable ML model training, evaluation, and inference in the cloud.
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.
Uphold rigorous standards for code correctness, security, and production-ready quality code.