Machine Learning Engineer Intern

To make electronics, you need chips, wafers, transistors, reticles, and... To make these, you must see, test and manufacture them at scale—faster and better than ever before. That's where KLA comes in. Whether you're early in your career or an experienced professional, you'll solve complex challenges, work alongside brilliant minds and help shape the future of technology.

The Semiconductor Products and Customers (Semi PC) business unit designs, builds and sells KLA’s product portfolio to help chip manufacturers meet high-quality standards for technologies like AI, data centers, automotive and electronic devices. Our inspection, metrology, and analytics systems detect and monitor defects across logic and memory chips, specialty semiconductors, wafers, reticles, advanced packaging, process equipment and materials. We also develop technologies for specialty processes, IC substrates, and PCB manufacturing, including etch and deposition, component inspection, and imaging and analytics.

The Film and Scatterometry Technology (FaST) division delivers industry-leading metrology solutions essential to advanced semiconductor manufacturing. Its portfolio includes systems for measuring film thickness, critical dimensions, composition, and resistivity—key to developing and scaling next-gen IC processes. FaST partners closely with customers from early R&D through high-volume production, ensuring performance and process alignment.

KLA is looking for a Machine Learning Engineer Intern to join our algorithms team. You'll work with experienced engineers and scientists to develop machine learning solutions for semiconductor manufacturing. You'll get hands‑on experience applying ML to real‑world data.

• 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.

• 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.

• Publications in physics‑informed machine learning.
• Experience with real‑world experimental or sensor data.
• Background in optics, imaging or other physical sciences.

• Requires less than 1 year of related experience.

We provide advanced inspection tools, metrology systems, process solutions, and computational analytics that make electronics possible, tackling complex challenges. From electron and photon optics to machine learning and data analytics, we seek perfection at the most fundamental level of matter in the universe. If you want to make electronics that push industries forward and make the world a better place, join us.

Total Rewards Base Pay Range: $49.00 - $58.00 per hour based on pursuit of a Masters and Ph.D.

Primary

Location:

USA-CA-Milpitas-KLA

• KLA’s total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to: medical, dental, vision, life, and 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…