Intermediate/ Senior Software Engineer - Cortex LLM Training Platform

Snowflake’s ML Platform team enables customers to run demanding machine learning and AI workloads inside Snowflake. The role focuses on designing and scaling the Cortex LLM post-training platform, including its APIs, SDKs, distributed systems, GPU infrastructure, and performance. The engineer will also productionize research techniques into reliable, composable components for enterprise-scale use.

Responsibilities:

• Design and build across the full stack — from the public training APIs and SDK through the control plane to the GPU data plane
• Scale the distributed systems that make GPU compute serverless — multi-tenant scheduling, placement, and capacity-aware routing across regional GPU pools, with fault tolerance built in
• Drive end-to-end performance at scale — keep the training, inference, and RL loops fast and the data plane responsive under heavy concurrent load, with GPUs kept saturated
• Productionize research building blocks — partner with Snowflake Research to turn state-of-the-art training and inference techniques into reliable, composable components customers can run at enterprise scale