Snowflake
We are seeking talented System Developers and Researchers to join the Snowflake AI Research team and contribute to LLM inference and training system development, optimizations, and agentic systems. We will push the boundaries of deep learning systems and drive cutting-edge innovations in AI.
Requirements
- Bachelor’s degree in Computer Science, Electrical Engineering, or a related field
- 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
- Generous Paid Time Off
- 401k Matching
- Retirement Plan
- Visa Sponsorship
- Four Day Work Week
- Generous Parental Leave
- Tuition Reimbursement
- Relocation Assistance
Originally posted on Himalayas
To apply for this job please visit himalayas.app.
Keep exploring on Get A Job.ai
Not quite the right fit? Your next opportunity is a click away.
- Browse all jobs
- More jobs by category
- Remote jobs you can do from anywhere
- Research typical pay for this role
- Set a job alert so new matches reach you first
- Upload your resume to apply faster
Hiring instead? Post a job and reach candidates searching right now.