AI System Research and Development Engineer - Optimization

We are seeking talented System Developers and Researchers to join the Snowflake AI Research team and contribute to LLM inference and training system development, optimizations, and agentic systems. We will push the boundaries of deep learning systems and drive cutting-edge innovations in AI.

Requirements

• Bachelor's degree in Computer Science, Electrical Engineering, or a related field
• 5 years of experience in GPU kernel optimization, deep learning system optimization, or high-performance computing (HPC)
• Proficiency in deep learning frameworks such as PyTorch, TensorFlow, JAX
• Strong understanding of GPU architectures and experience with CUDA or similar frameworks
• Experience with frameworks like CUTLASS, Triton, cuDNN, etc.
• Experience with profiling tools (e.g., nvprof, Nsight) and performance analysis methodologies
• Solid problem-solving skills and ability to debug complex performance issues
• Excellent communication skills and ability to work effectively in a cross-functional team environment

Benefits

• Generous Paid Time Off
• 401k Matching
• Retirement Plan
• Visa Sponsorship
• Four Day Work Week
• Generous Parental Leave
• Tuition Reimbursement
• Relocation Assistance