AI System Research and Development Engineer – Optimization

Snowflake

AI System Research and Development Engineer optimizing GPU performance for LLM tasks at Snowflake. Join a team innovating in AI and deep learning systems.

Posted 4/30/2026full-timeBellevue • California, Washington • 🇺🇸 United StatesMid-LevelSenior💰 $200,000 - $287,500 per yearWebsite

Tech Stack

Tools & technologies

PyTorchTensorflow

About the role

Key responsibilities & impact

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.

Requirements

What you’ll need

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.

Benefits

Comp & perks

Employees must abide by the company’s data security plan
Confidentiality and security standards for handling sensitive data

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

GPU kernel optimizationdeep learning system optimizationhigh-performance computingPyTorchTensorFlowJAXCUDACUTLASSTritoncuDNN

Soft Skills

problem-solvingcommunicationteamwork

Certifications

Bachelor’s degree in Computer ScienceBachelor’s degree in Electrical EngineeringMaster’s degreePhD