Staff Software Engineer - Machine Learning Platform

Snowflake•Menlo Park, CA

18d

About The Position

Snowflake is about empowering enterprises to achieve their full potential — and people too. With a culture that’s all in on impact, innovation, and collaboration, Snowflake is the sweet spot for building big, moving fast, and taking technology — and careers — to the next level. Build the future of data. Join the Snowflake team. The Snowflake Machine Learning Platform team’s mission is to enable customers to bring their machine learning and deep learning workloads to Snowflake. Our customers want to build powerful models with the ever-increasing data in Snowflake but face several challenges including infrastructure optimizations, orchestration, performance, and security. The team aims to solve these challenges by building highly integrated platform solutions that are simple, secure, and enable end-to-end ML workflows. We are on an early journey to build the most scalable machine learning and data platform without sacrificing the benefits of a single platform and governance. We are looking for outstanding technical leaders who will join our ML Platform team to build the next-generation platform and play a pivotal role in this journey by understanding Snowflake’s core platform architecture and evolving it to enable state-of-the-art machine learning and LLM workloads. Join us to define strategies, set technical directions, design and execute, engage and deliver innovation, and unlock the power of AI for thousands of enterprise customers. This position is based in Menlo Park, CA, and Bellevue, WA

Requirements

Have 7+ years of industry experience designing, building, and supporting Internet serving infrastructure, machine learning platforms, machine learning services, and frameworks.
Strong track record of working with machine learning systems and/or platforms.
Experience in serving LLMs using inference engines like vLLM, TensorRT-LLM, TEI, SGLang, and knowing tradeoffs between them.
Experience serving fine-tuned LLMs (PEFT, DPO, RL).
Have built a roadmap and vision around machine learning teams, and led technical decision making with help of architects and PMs and team.
BS/MS/PhD in Computer Science or related majors, or equivalent experience

Nice To Haves

Experience with several of the following frameworks: SKLearn, XGBoost, PyTorch, Tensorflow, MLflow is a plus.
Previous experience in building batch and real-time ML serving systems preferred.

Responsibilities

Help define and own the roadmap, working collaboratively and proactively with senior architects, PMs, and team leadership. The initiatives include platforms and tools that enable customers to do state-of-the-art machine learning on Snowflake natively.
Collaboratively build and execute a vision for incorporating new advances in machine learning in ways that best achieve the team’s business objectives.
Ensure operational excellence of the services and meet the commitments to our customers regarding reliability, availability, and performance.
Collaborate across other ML partner teams to continuously improve ML development velocity and capabilities at Snowflake.
Support team members in delivering a high level of technical quality.