Snowflake

AI System Research and Development Engineer - Optimization

Job Description

Posted on: 
2025-03-28

Responsibilities

  • Analyze and optimize GPU kernel performance for LLM training and inference.
  • Develop strategies to enhance efficiency and scalability of deep learning systems.
  • Profile and benchmark systems to identify performance bottlenecks.
  • Implement optimizations to reduce latency and improve resource utilization.
  • Stay updated with advancements in GPU kernel optimization and deep learning.
  • Contribute to the development of agentic frameworks for LLM-driven workflows.
  • Publish innovations and optimizations in technical blogs and conferences.

Job Requirements

  • Bachelor’s degree in Computer Science, Electrical Engineering, or related field; Master’s or PhD preferred.
  • 5 years of experience in GPU kernel optimization or deep learning system optimization.
  • Proficiency in deep learning frameworks (e.g., PyTorch, TensorFlow).
  • Strong understanding of GPU architectures and CUDA experience.
  • Experience with profiling tools and performance analysis methodologies.
  • Solid problem-solving skills for debugging performance issues.
  • Excellent communication skills for cross-functional teamwork.
Apply now

More job openings