Senior Software Engineer — LLM Post-Training Platform

Snowflake's ML Platform team is building Cortex Training, an LLM post-training platform that turns scarce GPU capacity into a simple, composable service. We are looking for a senior software engineer to design and build across the full stack, scale distributed systems, drive end-to-end performance, and productionize research building blocks.

• Design and build across the full stack — from public training APIs and SDK through the control plane to the GPU data plane. Scale distributed systems for GPU compute serverless with multi-tenant scheduling, placement, capacity-aware routing, and fault tolerance. Drive end-to-end performance at scale, keeping training, inference, and RL loops fast and GPUs saturated. Productionize research building blocks by partnering with Snowflake Research to turn state-of-the-art techniques into reliable, composable components.

• 5+ years building and shipping production ML systems. Strong distributed systems and infrastructure foundation — designing scalable, fault-tolerant services on Kubernetes. Familiarity with GPU and LLM infrastructure (PyTorch, DeepSpeed/FSDP, Ray, CUDA/NCCL, vLLM). Demonstrated ability to harden systems for reliability, throughput, and cost efficiency. BS in Computer Science or related field.

• Hands-on LLM post-training / modeling experience. MS/PhD in Computer Science or related field.