AI/ML Engineering Intern

KLA Corporation • Milpitas, CA

29d • $47 - $57 • Onsite

About The Position

Join the BBP Algorithm team that powers some of the world's most advanced optical patterned wafer inspection systems. You'll help push the state of the art in computer vision and deep learning to detect yield-limiting defects-improving sensitivity, robustness, and throughput on production tools used in leading fabs.

Requirements

• Currently pursuing MS or PhD in Computer Science, Electrical/Computer Engineering, Mathematics, Physics, or related field (exceptional BS candidates considered).
• Hands-on experience training deep learning models for vision (classification, detection, segmentation, or anomaly).
• Strong foundations in linear algebra, probability/statistics, and optimization; ability to analyze complex, noisy data.
• Proficiency with Python and modern DL frameworks (PyTorch or TensorFlow) on Linux; solid software fundamentals (Git, testing, code review).

Nice To Haves

• Knowledge of image processing / computer vision beyond DL (filters, registration, morphology, feature engineering).
• Experience with C++ in Linux/Windows and/or CUDA/GPU acceleration; familiarity with performance profiling and memory optimization.
• Exposure to MLOps practices (experiment tracking, model/data versioning, reproducible pipelines).
• Background or interest in semiconductor manufacturing (lithography, VLSI processes) is a plus.

Responsibilities

• Advance AI/ML models for defect detection, classification, and anomaly discovery on real wafer imagery; design experiments, define metrics, and drive SOTA baselines forward.
• Curate and engineer datasets (balancing/partitioning, augmentation strategies, labeling guidance) to maximize label efficiency and generalization across layers, technologies, and tools.
• Train, evaluate, and iterate using modern deep learning frameworks; build repeatable pipelines for preprocessing, training, inference, and reporting.
• Prototype algorithmic ideas (e.g., semi/self-supervised learning, domain adaptation, uncertainty estimation, active learning) and transition the best candidates into production-quality code.
• Optimize for scale and speed (efficient ops, quantization/pruning, GPU utilization) to meet on-tool and near-real-time constraints.
• Collaborate cross-functionally with system engineering and software teams on integration, testing, and demo/beta evaluations; contribute to documentation and quality practices.

Benefits

• KLA's total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to: medical, dental, vision, life, and 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 and paid company holidays, and family care and bonding leave.
• Interns are eligible for some of the benefits listed.

What This Job Offers

• Job Type: Full-time
• Career Level: Intern
• Industry: Computer and Electronic Product Manufacturing
• Number of Employees: 5,001-10,000 employees