# AI System Research and Development Engineer - Optimization

**Company:** [Snowflake](https://hotfix.jobs/companies/snowflake)
**Location:** Bellevue, WA
**Role:** ML Engineering
**Salary:** $200k – $265k/yr
**Experience:** 5+ years
**Skills:** PyTorch, TensorFlow, JAX, CUDA, Cutlass, Triton, Cudnn, Nvprof, Nsight, Gpu Kernel Optimization
**Posted:** 2026-06-16

> Develop and optimize GPU kernels and deep learning systems for LLM training and inference at Snowflake AI Research. Requires 5+ years in GPU/HPC optimization and strong proficiency in PyTorch, TensorFlow, JAX, and CUDA.

## Job Description

## 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

## Requirements
- Bachelor’s degree in Computer Science, Electrical Engineering, or a related field (Master’s degree or PhD 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

## Similar roles

- [Research Engineer](https://hotfix.jobs/jobs/b4d18018-e64d-4828-a8b4-303225328e17) - Console - San Francisco, CA - $200k – $350k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/1b64040d-77ad-441e-85bb-9de3583cf9b8) - Kepler - New York, NY - $200k – $280k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/bc31e7f5-d976-4a97-a1be-bee1a0b92f94) - AfterQuery - San Francisco, CA - $200k – $300k/yr
- [Machine-Learning Operations Engineer](https://hotfix.jobs/jobs/d80db2a3-5a47-41f4-b4ae-220a73b77cbf) - Tennr - New York, NY - $200k – $230k/yr
- [Machine Learning Engineer, Enterprise Brain](https://hotfix.jobs/jobs/b5ce4e0d-3ddb-4ef6-82aa-18b21d4639ff) - Glean - Mountain View, CA - $200k – $300k/yr

**Apply:** https://hotfix.jobs/jobs/5396c247-8435-4112-8bbc-81c2a8853542
**Canonical:** https://hotfix.jobs/jobs/5396c247-8435-4112-8bbc-81c2a8853542