AI GPU Arch Perf Optimization Intern

About the role

This internship focuses on optimizing GPU compute kernels for AI and numerical workloads, reproducing real inference and training workloads for hardware validation, and profiling performance bottlenecks to inform architecture design. Candidates need Python skills, solid AI fundamentals and basic computer systems knowledge, with GPU kernel or parallel computing exposure a plus.