← Back to Careers
Member of Technical Staff
Summer Internship•San Francisco, CA
Summer internship opportunity to contribute to our GPU optimization platform. You'll work on meaningful projects that directly impact our product while gaining hands-on experience with cutting-edge AI infrastructure.
What You'll Do
- • Develop tools for automated GPU performance optimization
- • Work on ML models for code generation and optimization
- • Contribute to our CUDA kernel optimization pipeline
- • Build visualization tools for performance analysis
- • Collaborate on research publications and blog posts
What We Look For
Core Technical Expertise
- • GPU Fundamentals: Deep understanding of GPU architectures, CUDA programming, and parallel computing patterns.
- • Deep Learning Frameworks: Proficiency in PyTorch, TensorFlow, or JAX, particularly for GPU-accelerated workloads.
- • LLM/AI Knowledge: Strong grounding in large language models (training, fine-tuning, prompting, evaluation).
- • Systems Engineering: Proficiency in C++, Python, and possibly Rust/Go for building tooling around CUDA.
Ideal Background
- • Publications or open-source contributions in GPU computing or ML/AI for code are a plus.
- • Hands-on experience with large-scale experiments, benchmarking, and performance tuning.