ML Engineer Intern - Physics-Informed Metrology for Chips

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

Job summary

KLA is seeking a Machine Learning Engineer Intern to join our algorithms team in Milpitas. You’ll develop ML solutions for semiconductor manufacturing, working with experienced engineers and scientists to apply ML to real‑world data.

Qualifications include pursuing a Master’s or PhD in a STEM field, strong Python and PyTorch experience, and solid ML fundamentals. This internship offers exposure to cutting‑edge metrology and data analytics within our Semi PC division.

Qualifications

• Currently pursuing a Master's or PhD in Computer Science, Electrical Engineering, Physics, Applied Mathematics or a related STEM field.
• Strong Python and hands‑on deep learning experience (PyTorch or similar).
• Solid ML and data science fundamentals and strong problem‑solving skills.

Responsibilities

• Build, train and evaluate ML and deep learning models for critical dimension (CD) metrology.
• Develop physics‑informed ML approaches that bring physical models and constraints into model design and training.
• Explore methods for learning from limited data and estimating uncertainty.
• Write clean, reusable Python code and present your results to the team.

Skills

Python

PyTorch

Deep learning

ML fundamentals

Education

Master's or PhD in CS/EE/Physics/Math

Job description

KLA is seeking a Machine Learning Engineer Intern to join our algorithms team in Milpitas. You’ll develop ML solutions for semiconductor manufacturing, working with experienced engineers and scientists to apply ML to real‑world data.

Qualifications include pursuing a Master’s or PhD in a STEM field, strong Python and PyTorch experience, and solid ML fundamentals. This internship offers exposure to cutting‑edge metrology and data analytics within our Semi PC division.