Snowflake
AI System Research and Development Engineer - Optimization
US-WA-Bellevue$200k–$265kfulltimemidAdded 2 days ago
About this role
Snowflake is seeking an AI System Research and Development Engineer focused on optimizing deep learning systems and large language models (LLMs). This role involves collaboration in a fast-paced environment to enhance system performance and scalability while driving innovation in AI.
What you'll do
- Analyze and optimize GPU kernel performance for LLM training and inference
- Develop strategies for efficiency and scalability of deep learning systems
- Profile and benchmark systems to identify bottlenecks
- Implement optimizations to reduce latency and improve resource use
- Contribute to the development of agentic frameworks for LLM workflows
- Publish innovations and optimizations through open-source channels
What they're looking for
- Deep learning frameworks (PyTorch, TensorFlow, JAX)
- GPU architectures and CUDA
- Performance profiling tools (nvprof, Nsight)
- Optimization using CUTLASS, Triton, cuDNN
- Strong problem-solving capabilities
- Excellent communication skills
- Ability to work in cross-functional teams
- Experience in high-performance computing (HPC)
Benefits
- Dynamic work environment
- Opportunities for innovation
- Collaboration with top AI professionals
- Growth opportunities within a fast-scaling company
- Supportive team culture
- Open-source contributions
Opens the official application on the employer’s site. No login required.