Staff Machine Learning Engineer (Generative AI Modeling and Inference)

• Snap’s Generative ML Platform team builds cutting-edge AI technologies that power creative, scalable experiences for hundreds of millions of Snapchatters worldwide. From multimodal LLMs and video generation to real-time AR, human understanding, and 3D content creation, we develop the full stack of generative AI, including foundational models, efficient infrastructure, and on-device and server-side inference
• Our team creates intuitive tools, platforms, and agentic systems that empower creators, developers, and internal teams to bring ideas to life, while advancing personalized, human-centric experiences across mobile, web, and wearable devices like Spectacles
• We’re looking for a Machine Learning Engineer to join Snap Inc!
• Develop innovative machine learning technology and products that serve millions of Snapchatters
• Build cutting-edge augmented reality experiences using generative models
• Deliver generative machine learning experiences on device
• Partner with cross-functional Snap teams to explore and prototype new products

Benefits

• Up to 28 weeks of parental leave and return-to-work program
• Adoption, surrogacy, infertility, and fertility preservation benefits
• Backup child care coverage, center-based care support, & caregiver assistance
• Dependent Daycare Flexible Spending Account
• Lactation rooms (based on office location) and breastmilk shipping coverage
• Short-term disability, long-term disability, basic life insurance, and AD&D insurance
• Services to support your path to parenthood & beyond - fertility support & family planning via Carrot, SNOO, Hinge Health
• For Snap, employees taking care of loved ones with complex, chronic, or ongoing care needs (i.e. parent with Dementia, spouse with Cancer, child with Autism), your family can save time, money, & stress with Wellthy
• Comprehensive medical coverage, including PPO, HDHP with HSA, and HMO options
• Dental coverage, including orthodontia benefits
• Vision coverage, including frame and contact allowance and LASIK benefits
• Free membership to One Medical for primary and urgent healthcare services
• Business Travel Insurance
• Gym perks and discounts
• Team fitness classes and races
• Sports leagues
• Cooking and nutritional workshops
• Well-being reimbursement
• Unlimited access to virtual physical therapy with Anthem
• 15 sessions per year of mental health for you + dependents through Lyra
• Generous time off and leave programs. In the US, this includes the following:
• Up to 3 weeks (120 hours) of vacation time and 1 floating holiday for non-exempt team members
• 15 sick and safe days per year
• 12 paid holidays per year
• Social gatherings, team outings, and volunteering programs
• Speaker series, classes, and subscriptions to educational programs
• Meditation and yoga classes
• Snap Inc. provides a 401(k) plan that allows you to save on a pre-tax, Roth, and after-tax basis for your retirement (yes, we even have the Mega backdoor option!) Snap will also match 100% up to 3% of your contribution and then 50% on your 4th and 5th%
• Flexible Spending Accounts for healthcare, daycare and commuter
• Health Savings Account with employer contributions
• Rocket Lawyer memberships
• Financial education programs
• Snap exclusive discounts and perks
• Compensation packages that let you share in Snap’s long-term success!- Desire to grow professionally, learn and help others
• Desire to solve open ambiguous problems
• Ability to effectively collaborate with internal teams and external partners
• Ability to work independently
• Deep understanding of mathematics and/or machine learning algorithms
• Familiarity with Neural Networks and Deep Learning and Generative Modeling
• A proven passion for machine learning; you stay up-to-date with research and are excited about prototyping new ideas quickly
• Bachelor’s degree in technical field such as computer science, mathematics, statistics or equivalent years of experience
• Experience with Computer Vision or Generative Modeling techniques
• Experience working with machine learning frameworks such as TensorFlow, PyTorch, JAX, MLX, scikit-learn, or related frameworks
• 8+ years of post-Bachelor’s machine learning or related experience; or a Master’s degree in a technical field + 7+ years of post-grad ML or related experience; or a PhD in a related technical field + 4+ years of post-grad ML or related experience
• Experience building and optimizing ML inference pipelines
• Knowledge of GPU, CPU, or NPU optimization techniques
• Knowledge of distillation, quantization and model compression techniques
• Advanced degree in computer science or related field
• Experience training large-scale diffusion models for images, videos or 3D