H1BConnect Pro is launching with premium alerts and access to more job postings.Get early access
KLA logo

AI/ML Engineering Intern

KLA
Milpitas, CA Internship 11/26/2025
Master's InternshipPhD Internship

Job Description

Join KLA's BBP Algorithm team as a full-time summer intern, where you'll advance AI/ML models for defect detection and improve production tools used in semiconductor manufacturing through cutting-edge computer vision and deep learning techniques.

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
  • Proficiency with Python and modern DL frameworks (PyTorch or TensorFlow) on Linux
  • Solid software fundamentals (Git, testing, code review)

Responsibilities

  • Advance AI/ML models for defect detection, classification, and anomaly discovery on real wafer imagery
  • Curate and engineer datasets to maximize label efficiency and generalization
  • Train, evaluate, and iterate using modern deep learning frameworks
  • Prototype algorithmic ideas and transition the best candidates into production-quality code
  • Optimize for scale and speed to meet on-tool and near-real-time constraints
  • Collaborate cross-functionally with system engineering and software teams