Deep Learning System Optimization Engineer

Build the future of the AI Data Cloud. Join the Snowflake team.

We are seeking talented System Developers and Researchers to join our Snowflake AI Research team and work on GPU kernel optimizations and deep learning system optimizations. Our focus is on building the most efficient and scalable deep learning systems. This is an exciting opportunity to collaborate with a strong team, including founding members of DeepSpeed, vLLM, and TensorFlow. Together, we will innovate, build, and advance the state of the art in deep learning systems.

Responsibilities:

• Analyze and optimize GPU kernel performance for training and inference of deep learning models.

• 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.

• Publish their innovations, optimizations, and engineering practices in technical blogs, top-tier conferences and journals.

Requirements:

• 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 anessential part of their duties. It is every employee's duty to keep customer information secure andconfidential.

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?

The following represents the expected range of compensation for this role:

• The estimated base salary range for this role is $195,000 - $287,500.
• Additionally, this role is eligible to participate in Snowflake's bonus and equity plan.

The successful candidate's starting salary will be determined based on permissible, non-discriminatory factors such as skills, experience, and geographic location. This role is also eligible for a competitive benefits package that includes: medical, dental, vision, life, and disability insurance; 401(k) retirement plan; flexible spending & health savings account; at least 12 paid holidays; paid time off; parental leave; employee assistance program; and other company benefits.

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?