Software Intern (GPU Optimization) at KLA

Milpitas, California, United States

KLA Logo
Not SpecifiedCompensation
InternshipExperience Level
Full TimeJob Type
UnknownVisa
Semiconductor, Electronics ManufacturingIndustries

Requirements

  • 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 tools

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/CUDA 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

Skills

GPU Optimization
Software Development
C++
Python
Semiconductor
Metrology
Inspection Systems
Wafer Inspection
Optical Systems
Process Control

KLA

Provides process control and yield management solutions

About KLA

KLA provides process control and yield management solutions primarily for semiconductor manufacturers. The company offers advanced inspection tools, metrology systems, and computational analytics that help manufacturers identify and fix defects during production. This process enhances the quality and reliability of electronic devices, leading to higher production yields. KLA distinguishes itself from competitors by focusing on high-precision equipment and software that are essential for defect detection in semiconductor manufacturing. The company's goal is to promote sustainability by committing to using 100% renewable electricity in its operations by 2030.

Milpitas, CaliforniaHeadquarters
1975Year Founded
IPOCompany Stage
Industrial & Manufacturing, EnergyIndustries
5,001-10,000Employees

Benefits

Health Insurance
Dental Insurance
Vision Insurance
Life Insurance
401(k) Retirement Plan
401(k) Company Match
Employee Stock Purchase Plan
Student Loan Assistance
Tuition Reimbursement
Wellness Program
Mental Health Support
Paid Vacation
Paid Holidays
Parental Leave

Risks

Emerging competition in solid-state battery technology may impact market share.
Rapid innovation in semiconductor processes may outpace KLA's current technology.
Potential delays in achieving renewable electricity goals could affect brand reputation.

Differentiation

KLA specializes in advanced inspection tools and metrology systems for semiconductors.
The company integrates computational analytics to enhance defect detection and yield management.
KLA is committed to sustainability, aiming for 100% renewable electricity by 2030.

Upsides

Rising demand for advanced inspection tools due to AI and IoT growth.
AI-driven predictive maintenance reduces downtime in semiconductor manufacturing.
Collaborations with cloud providers enhance data analytics and process optimization.

Land your dream remote job 3x faster with AI