Software Intern (GPU Optimization) at KLA in Milpitas, CA

Summary: KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. They are seeking a summer intern to join their advanced wafer inspection algorithm team, focusing on GPU acceleration and CUDA code optimization for high-throughput image processing and machine learning workflows. Responsibilities: - Design, implement, and optimize CUDA kernels for high-throughput image analysis and defect detection - Profile and tune GPU workloads using NVIDIA Nsight Compute, Nsight Systems, and other performance tools - Collaborate with algorithm engineers to integrate GPU/CUA optimized modules into inspection pipelines - Explore advanced optimization techniques such as memory optimization, thread and block configuration optimization, instruction-level optimization, kernel fusion and launch overhead reduction, algorithmic optimization and hardware-specific tuning - Analyze performance bottlenecks and propose architectural improvements - Document technical findings and present results to cross-functional teams Required Qualifications: - Currently enrolled in a Master or PhD program in Computer Science, Electrical Engineering, Applied Physics, or a related field - Strong proficiency in CUDA, C/C++, and Python - Deep understanding of GPU architecture, memory hierarchy, and parallel computing principles - Demonstrated expertise in GPU performance analysis and optimization using advanced profiling tools such as NVIDIA Nsight Compute, Nsight Systems, and CUDA Visual Profiler, with a proven ability to identify and resolve memory bottlenecks and kernel inefficiencies - Background in image processing, computer vision, or machine learning - Strong analytical and problem-solving skills - Ability to work independently and collaboratively in a fast-paced environment - Excellent communication and presentation skills - Requires less than 1 year of related experience Required Skills: CUDA, C/C++, GPU optimization Important Skills: Python, Image processing, Machine learning Nice-to-Have Skills: Analytical skills, Problem-solving skills, Communication skills, Team collaboration Benefits: Medical, Dental, Vision, Life, Other voluntary benefits, 401(K) including company matching, Employee stock purchase program (ESPP), Student debt assistance, Tuition reimbursement program, Development and career growth opportunities and programs, Financial planning benefits, Wellness benefits including an employee assistance program (EAP), Paid time off, Paid company holidays, Family care and bonding leave