Software Engineering Intern – AI/ML for Diagnostics & Calibration - Milpitas, CA, United States

Originally posted 23 October 2026 by the employer.

About the role

KLA Corporation is seeking a Software Engineering Intern to develop AI/ML prototypes and analytical models for diagnostics and calibration. This role sits within the RAPID division, which focuses on reticle inspection solutions for mask shops and fabs.

What you'll do

- Develop Python-based software and AI/ML prototypes that combine analytical models with image-based diagnostics. - Prepare representative image and signal datasets and assess metrics including resolution, noise, spectral content, sampling and aliasing, uniformity and contrast, and optical or system aberrations. - Evaluate prototypes quantitatively against analytical baselines and expert assessments, documenting accuracy, uncertainty, and failure cases. - Identify performance-limiting parameters and likely contributors to degradation, and recommend calibration actions for human review. - Build reproducible code, tests, and technical documentation while collaborating with engineers using approved data and AI tools. - Explore AI-assisted troubleshooting, engineering analysis, or workflow automation with a mentor to support core diagnostics work.

What you'll need

- Current enrollment in a BS, MS, or PhD program in computer science, electrical engineering, applied mathematics, physics, or a related discipline. - Proficiency in Python and scientific computing. - Foundational knowledge of machine learning. - Experience analyzing data through coursework, research, or projects.

Nice to have

- Exposure to image or signal processing, optics, imaging systems, or calibration methods. - Experience with PyTorch, TensorFlow, or scikit-learn. - Familiarity with software testing, version control, or reproducible experimentation. - Familiarity with LLMs, retrieval-augmented generation, AI coding assistants, or workflow automation.

Skills: Python, Machine Learning, Image Processing, Diagnostics, Calibration