AI/ML Imaging Software Intern for Chip Diagnostics

KLA; Milpitas (CA); On-site; USD 51,000 - 80,000; Full time

Benefits offered by this job

Medical benefits

401(k) matching

Employee stock purchase plan

Tuition reimbursement

Paid time off

Job summary

KLA Corporation in Milpitas, CA, offers an internship focusing on Python-based software and AI/ML prototypes for image-based diagnostics. You will prepare datasets, evaluate metrics, and document results while collaborating with engineers to refine core diagnostics workflows.

The role emphasizes machine learning fundamentals, data analysis, and effective communication, with exposure to cutting-edge imaging systems and calibration methods. Interns eligible for associated benefits.

Qualifications

• Currently enrolled 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, with foundational knowledge of machine learning.
• Experience analyzing data through coursework, research, or projects; strong problem-solving skills and clear written and verbal communication.

Responsibilities

• 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: resolution, noise, spectral content, sampling and aliasing, uniformity and contrast, and optical or system aberrations.
• Evaluate prototypes quantitatively against analytical baselines and document 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; collaborate with engineers using approved data and AI tools while protecting proprietary information.
• Explore AI-assisted troubleshooting, engineering analysis, or workflow automation with your mentor to support the core diagnostics work.

Skills

Python

Data analysis

Problem solving

Effective communication

Education

BS/MS/PhD in CS/EE/Math/Physics

Tools

PyTorch

TensorFlow

scikit-learn

Testing

Version control

Job description

KLA Corporation in Milpitas, CA, offers an internship focusing on Python-based software and AI/ML prototypes for image-based diagnostics. You will prepare datasets, evaluate metrics, and document results while collaborating with engineers to refine core diagnostics workflows.

The role emphasizes machine learning fundamentals, data analysis, and effective communication, with exposure to cutting-edge imaging systems and calibration methods. Interns eligible for associated benefits.