Principal Software Engineer - Machine Learning

About the position

Build the future of machine learning within Snowflake! We're building cutting-edge ML infrastructure and AI-powered agents to empower users to effortlessly build and deploy powerful models directly on their data. Join us to solve real-world ML challenges at scale and deliver a massive leap in ML developer experience. We aim to build highly integrated products that are simple, secure, scalable and enable end to end ML workflows. We deliver AI powered ML tooling, highly optimized runtimes that power interactive and batch ML workloads for both structured and unstructured data, offline/online feature stores and MLOps capabilities. We are seeking a highly experienced and visionary Principal Engineer to lead our efforts in revolutionizing data scientist/MLE productivity through the strategic application of Generative AI to machine learning workflows. This is a unique opportunity to shape the future of data science and ML by building cutting-edge agentic tools and platforms that empower our customers to innovate faster and more efficiently. You will work closely with cross-functional teams including product and research, to identify key challenges and develop innovative solutions that challenge the status quo in the industry.

Responsibilities

• Build customer facing AI based services and platforms for machine learning.
• Design and implement highly scalable distributed platforms within the global Snowflake platform.
• Participate in decision-making processes on technical or business issues.
• Collaborate with engineers across teams to help deliver cross-functional initiatives.
• Ensure operational readiness of the services and meet the commitments to our customers regarding reliability, availability and performance.

Requirements

• 10+ years experience building large-scale AI/ML systems. Proven leadership as a Principal Engineer focused on ML developer tools and platforms.
• Passionate about AI/ML infrastructure and applications of LLMs.
• Deep multi-faceted appreciation in building end-to-end customer facing products. Experience working with cross teams, research teams and product managers.
• Strong CS fundamentals, including proficiency with data structures, algorithms, and distributed systems (Fluency in Python, or Java preferred).
• Familiarity or (preferred) expertise with machine learning and deep learning frameworks like Pytorch, XGBoost, etc.
• BS/MS/PhD in Computer Science or related majors.

Benefits

• Medical, dental, vision, life, and disability insurance
• 401(k) retirement plan
• Flexible spending & health savings account
• At least 12 paid holidays
• Paid time off
• Parental leave
• Employee assistance program
• Other company benefits