Intermediate/ Senior Software Engineer - Cortex LLM Training Platform

📋 Description

• Design and build across the full stack — from the public training APIs and SDK through the control
• Scale the distributed systems that make GPU compute serverless — multi-tenant scheduling
• Drive end-to-end performance at scale — keep the training, inference, and RL loops fast and the
• Productionize research building blocks — partner with Snowflake Research to turn state-of-the-art

🎯 Requirements

• 3+ years (Intermediate) | 6+ years (Senior) building and shipping production ML systems
• Strong distributed systems and infrastructure foundation — designing scalable, fault-tolerant
• Familiarity with GPU and LLM infrastructure — e.g., PyTorch, DeepSpeed/FSDP, Ray, CUDA/NCCL, vLLM
• Demonstrated ability to harden complex systems for reliability, throughput, and cost efficiency.
• BS in Computer Science or a related field (MS/PhD a plus).
• (Bonus) Hands-on LLM post-training / modeling experience — the strongest candidates pair deep infra

🎁 Benefits

• Snowflake is growing fast and scaling our team to accelerate growth.
• Opportunity to work with a team that ships fast and values reliability and performance.
• Collaborate with Snowflake Research to bring state-of-the-art training techniques into