companyWafer logo

Technical Internship in AI and GPU Optimization

WaferSan Francisco
On-site Internship

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Entry Level

Qualifications

What We Look ForWe are seeking candidates who have a blend of technical skills and a passion for innovation in AI and GPU technologies. Your ability to collaborate with a team and contribute to complex projects will be key to your success.

About the job

About the Role

We're excited to invite you to join wafer as a Spring Intern, where you will play a crucial role in shaping the future of AI infrastructure and GPU optimization. As part of our innovative team, you will work closely with full-time engineers to define our technical strategies and contribute to the development of the essential systems that drive our GPU optimization platform.

Your Responsibilities

  • Design and implement scalable infrastructure for AI model training and inference tasks.

  • Guide the team in making technical decisions and architectural choices.

Qualifications We Seek

Essential Technical Skills

  • GPU Fundamentals: A strong grasp of GPU architectures, CUDA programming, and parallel computing methodologies.

  • Deep Learning Frameworks: Skilled in PyTorch, TensorFlow, or JAX, especially for GPU-accelerated applications.

  • Knowledge of LLM/AI: Solid foundation in large language models, including training, fine-tuning, prompting, and evaluation.

  • Systems Engineering: Proficient in C++, Python, and potentially Rust/Go for developing tools around CUDA.

Preferred Background

  • Publications or contributions to open-source projects related to inference GPU computing or ML/AI are advantageous.

  • Hands-on experience in conducting large-scale experiments, benchmarking, and performance optimization.

About Wafer

wafer is at the forefront of AI infrastructure development, dedicated to pushing the boundaries of GPU optimization. We foster a collaborative environment where innovation thrives, and we are passionate about empowering our team to make impactful contributions.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.