KLA · 1 month ago
AI/ML Engineering Intern
Milpitas, CA
Internship
Onsite
Intern
$47/hr - $57/hr
1+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. The AI/ML Engineering Intern will join the BBP Algorithm team to advance AI/ML models for defect detection and collaborate cross-functionally on integration and testing.
ElectronicsInformation TechnologyManufacturing
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
Qualification
Required
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)
Requires less than 1 year of related experience
Preferred
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
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
Company
KLA
Kla creates tools and services that promote innovation in the electronics industry.
10001+ employees
H1B Sponsorship
KLA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)
Funding
Current Stage
Late StageLeadership Team
Lisa Grunwald
FISH Philosophy CEO
Rick Wallace
President & CEO
Recent News
news.com.au — Australia’s leading news site for latest headlines
2025-08-01
2025-08-01
Company data provided by crunchbase