Overview
On Site
USD 195,000.00 - 287,500.00 per year
Full Time
Skills
Research and Development
Cloud Computing
Research
Generative Artificial Intelligence (AI)
Collaboration
Scalability
Training
Workflow
Reasoning
Decision-making
Open Source
Computer Science
Electrical Engineering
Optimization
High Performance Computing
HPC
PyTorch
TensorFlow
JAX
GPU
CUDA
Performance Analysis
Problem Solving
Conflict Resolution
Debugging
Communication
Cross-functional Team
Deep Learning
Artificial Intelligence
Data Security
Innovation
Snow Flake Schema
Job Details
Build the future of the AI Data Cloud. Join the Snowflake team.
We are looking for 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. Our mission is to build the most efficient and scalable generative AI systems.
Recent releases from our team include SwiftKV, an advanced inference optimization, and Arctic LLM, one of the largest open-source MoE foundation models. This is an exciting opportunity to collaborate with a world-class team, including founding members of DeepSpeed, vLLM, and TensorFlow. Together, we will push the boundaries of deep learning systems and drive cutting-edge innovations in AI.
Responsibilities:
Join us in optimizing deep learning systems and pushing the boundaries of AI efficiency. Apply now to be part of our dynamic and pioneering team!
Every Snowflake employee is expected to follow the company's confidentiality and security standards for handling sensitive data. Snowflake employees must abide by the company's data security plan as an essential part of their duties. It is every employee's duty to keep customer information secure and confidential.
Snowflake is growing fast, and we're scaling our team to help enable and accelerate our growth. We are looking for people who share our values, challenge ordinary thinking, and push the pace of innovation while building a future for themselves and Snowflake.
How do you want to make your impact?
For jobs located in the United States, please visit the job posting on the Snowflake Careers Site for salary and benefits information: careers.snowflake.com
We are looking for 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. Our mission is to build the most efficient and scalable generative AI systems.
Recent releases from our team include SwiftKV, an advanced inference optimization, and Arctic LLM, one of the largest open-source MoE foundation models. This is an exciting opportunity to collaborate with a world-class team, including founding members of DeepSpeed, vLLM, and TensorFlow. Together, we will push the boundaries of deep learning systems and drive cutting-edge innovations in AI.
Responsibilities:
- Analyze and optimize GPU kernel performance for training and inference of LLMs.
- Develop and implement strategies to enhance the efficiency and scalability of deep learning systems.
- Profile and benchmark deep learning systems using tools and techniques to identify bottlenecks.
- Design and implement optimizations to reduce latency and improve resource utilization for training and inference.
- Stay updated with the latest advancements in GPU kernel optimization, deep learning, and LLM system development.
- Contribute to the development of agentic frameworks and applications for LLM-driven workflows, enhancing automation, reasoning, and decision-making capabilities.
- Open-source and publish innovations, optimizations, and engineering practices in technical blogs, top-tier conferences and journals.
- Bachelor's degree in Computer Science, Electrical Engineering, or a related field. A Master's degree or PhD is preferred.
- 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.
Join us in optimizing deep learning systems and pushing the boundaries of AI efficiency. Apply now to be part of our dynamic and pioneering team!
Every Snowflake employee is expected to follow the company's confidentiality and security standards for handling sensitive data. Snowflake employees must abide by the company's data security plan as an essential part of their duties. It is every employee's duty to keep customer information secure and confidential.
Snowflake is growing fast, and we're scaling our team to help enable and accelerate our growth. We are looking for people who share our values, challenge ordinary thinking, and push the pace of innovation while building a future for themselves and Snowflake.
How do you want to make your impact?
For jobs located in the United States, please visit the job posting on the Snowflake Careers Site for salary and benefits information: careers.snowflake.com
Employers have access to artificial intelligence language tools (“AI”) that help generate and enhance job descriptions and AI may have been used to create this description. The position description has been reviewed for accuracy and Dice believes it to correctly reflect the job opportunity.