Member Of Technical Staff (Winter Intern)
Wafer · San Francisco, CA · 9 mo ago
On-siteEngineering$6k–$10k/moInternship
About the role
You'll work directly with the team to define our technical direction and build the core systems that power our GPU optimization platform.
Responsibilities
- Build scalable infrastructure for AI model training and inference
- Lead technical decisions and architecture choices
Requirements
- 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.
- Publishations or open-source contributions in inference GPU computing or ML/AI for code are a plus.
- Hands-on experience with large-scale experiments, benchmarking, and performance tuning.
Qualifications
Any relevant experience is considered.
Skills
- C++, Python, TypeScript, Torch/PyTorch, CUDA
Benefits
Will sponsor visa.
Pay
$6K - $10K / monthly
Schedule
Not specified
Benefits
Will sponsor visa.