Padmi
Wafer logo
Wafer

GPU optimization · LLM inference

Member Of Technical Staff (Summer Intern)

San Francisco Bay Area · Onsite$6k–$10k/moPosted 7 days ago
Software engineeringInternInternshipSponsorship mentioned
Apply at Wafer

Opens the source posting on ycombinator.com

Source description

About the role

View original

Join our team to build the future of inference, GPU optimization and AI infrastructure. You'll work directly with the team to define our technical direction and build the core systems that power our GPU optimization platform.

What You'll Do

  • Build scalable infrastructure for AI model training and inference

  • Lead technical decisions and architecture choices

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.

More at Wafer

Related open roles

View all roles