Intermediate/Senior Software Engineer
About the role
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
Responsibilities
- Design and build across the full stack from public training APIs and SDK to GPU data plane
- Scale distributed systems for serverless GPU compute with fault tolerance
- Drive end-to-end performance at scale for training and inference
- Productionize research techniques into reliable components for enterprise use
Qualifications
- The role requires expertise in distributed systems, GPU infrastructure, and scaling machine learning workloads.
- Candidates should be able to productionize research techniques into reliable, enterprise-scale components.
Skills mentioned
About Jobverse.io
Go from 'Job Found' to 'Applied' in under 60 seconds with Jobverse.io.