Snap Inc. is looking for a Principal Machine Learning Engineer to join their ML Inference Platform team to design, implement, and scale critical machine learning components and services that support Snap's most strategic initiatives and push the limits of what's possible with machine learning.
Requirements
- Experience with GPU/TPU inference and optimizations
- Experience with TensorFlow, PyTorch, or related deep learning frameworks
- Experience with Docker, Kubernetes, Ray, NoSQL solutions, Memcache/Redis, Google/AWS services
- Experienced in MLOps and managing production machine learning lifecycle
- Strong understanding of machine learning approaches and algorithms
- Excellent programming and software design skills, including debugging, performance analysis, and test design
- Proven track record of operating highly-available systems at scale
Responsibilities
- Design, implement, and scale critical machine learning components and services to support Snap's most strategic initiatives
- Design and build a next-generation inference framework and services that can support large-scale model, high-throughput serving, enabling us to push the limits of what's possible with machine learning
- Perform model and inference optimization with various GPUs to improve model inference speed and efficiency
- Work across teams to understand product requirements, evaluate trade-offs, and deliver the solutions needed to build innovative products or services
- Advocate for and apply best practices when it comes to availability, scalability, operational excellence, and cost management
- Provide technical direction that influences the entire company
Other
- 2+ years of experience as a technical lead
- Ability to proactively learn new concepts and technology and apply them at work
- Skilled at solving ambiguous problems
- Strong collaboration and mentorship skills
- Experience leading teams and driving technical roadmaps
