Location: Remote (Global)
Posted: February 1, 2025
About Snowflake
At Snowflake, we believe in building a future where AI efficiency is the backbone of our data cloud. We're on a mission to transform the way data is stored, processed, and analyzed, and we're looking for the brightest minds to help us build the future of the AI Data Cloud. Our AI Research team is growing fast, and we’re on the lookout for a Deep Learning System Optimization Engineer to help us take our systems to the next level!
In this role, you'll join an innovative team, working alongside founding members of DeepSpeed, vLLM, and TensorFlow—industry leaders who have redefined what's possible in deep learning. If you're passionate about GPU kernel optimization, deep learning systems, and scalable AI architectures, this is the role for you!
What You’ll Do
? Optimize GPU Kernel Performance
- Work on the training and inference of deep learning models, focusing on optimizing GPU kernel performance to achieve faster and more efficient computations.
???? Enhance Deep Learning Systems
- Implement strategies to increase the scalability and efficiency of deep learning systems, ensuring that Snowflake can process massive datasets with minimal latency.
????????? Profile & Benchmark Systems
- Use cutting-edge tools and techniques to identify bottlenecks in deep learning systems, providing detailed performance analysis to ensure top-tier optimization.
???? Develop High-Performance Solutions
- Design and implement system optimizations to reduce latency, improve resource utilization, and enhance the overall performance of Snowflake’s AI infrastructure.
???? Innovate & Share
- Stay at the forefront of GPU kernel optimization and deep learning technology. Publish your innovations and engineering practices in top-tier conferences, journals, and technical blogs.
What We’re Looking For
???? Academic & Professional Excellence
- Bachelor’s degree in Computer Science, Electrical Engineering, or related fields (a Master’s or PhD is highly preferred).
- 5+ years of experience in GPU kernel optimization, deep learning system optimization, or high-performance computing (HPC).
???? Deep Learning Framework Proficiency
- Experience with frameworks like PyTorch, TensorFlow, and JAX.
- Familiarity with CUTLASS, Triton, cuDNN, and other relevant tools.
? Advanced GPU Knowledge
- Strong understanding of GPU architectures and experience with CUDA (or similar frameworks).
- Proficiency with profiling tools such as nvprof, Nsight, and other performance analysis methodologies.
???? Problem-Solving Skills
- A knack for debugging complex performance issues and developing high-impact solutions that push the boundaries of AI efficiency.
???? Collaboration & Communication
- Excellent communication skills and a collaborative mindset. Ability to work effectively across teams to solve problems and achieve project goals.
Why Snowflake?
???? Innovative & Collaborative Environment
- Work alongside some of the brightest minds in AI, deep learning, and high-performance computing, including leading experts in the field.
???? Scale Your Impact
- As Snowflake grows, you'll play a key role in scaling our AI systems, helping our customers unlock the full potential of their data.
???? Global & Remote Team
- Join a diverse and global team that values creative problem solving, continuous learning, and pushing the boundaries of what's possible.
???? Pioneering AI Excellence
- Help us build the future of AI—where our deep learning systems lead the way in efficiency, scalability, and groundbreaking innovation.
Ready to Make an Impact?
Join us at Snowflake and help us optimize the deep learning systems that power the next era of AI. Apply now and be part of a pioneering team that’s shaping the future of AI and the AI Data Cloud. We’re scaling fast, and we want you to be part of our journey!
