AI Research Engineer (LLM)

TL;DR

AI Research Engineer (LLM): Developing and optimizing high-performance systems for LLM inference and training with an accent on GPU kernel performance and deep learning scalability. Focus on designing efficient inference optimizations, profiling complex bottlenecks, and contributing to agentic frameworks for LLM-driven workflows.

Location: Must be based in Bellevue, WA, USA

Salary: $200,000 – $265,000

Company

Snowflake is a cloud-based data platform company powering the era of the agentic enterprise through innovative AI and data solutions.

What you will do

• Analyze and optimize GPU kernel performance for LLM training and inference.
• Develop strategies to enhance the efficiency and scalability of deep learning systems.
• Profile and benchmark systems to identify and resolve performance bottlenecks.
• Design optimizations to reduce latency and improve resource utilization.
• Contribute to the development of agentic frameworks for LLM-driven workflows.
• Publish innovations and engineering practices in technical blogs and top-tier conferences.

Requirements

• Bachelor’s degree in Computer Science or related field (Master’s or PhD preferred).
• 5+ years of experience in GPU kernel optimization, deep learning systems, or HPC.
• Proficiency in deep learning frameworks like PyTorch, TensorFlow, or JAX.
• Strong understanding of GPU architectures and experience with CUDA.
• Experience with frameworks such as CUTLASS, Triton, or cuDNN.
• Experience with profiling tools like nvprof or Nsight.

Culture & Benefits

• Opportunity to work on cutting-edge generative AI and agentic systems.
• Collaborative environment focused on innovation and experimental mindset.
• Access to high-impact projects with global scale.
• Commitment to open-source contributions and technical research.