AI Engineer - Vision & Defect Detection for Semiconductors

KLA; Milpitas (CA); On-site; USD 136,000 - 232,000; Full time

Benefits offered by this job

401(K) matching

Employee stock purchase program (ESPP)

Tuition reimbursement

Job summary

KLA is seeking a motivated AI Engineer for its RAPID division in Milpitas, CA. You will design, train, and deploy deep learning models blended with physics-based methods to advance KLA's advanced reticle inspection platforms. This role requires collaboration across teams, dataset curation, and delivering production‑quality code.

International travel may be involved. Candidates should have a PhD (or 1–2 years) in Applied Mathematics, CS, EE or related field, strong CV/DL experience, proficiency

Qualifications

• Publications listed on resume required.
• Doctorate degree; 0 yrs experience or Master’s 3 yrs or Bachelor’s 5 yrs related work experience.

Responsibilities

• Design, train, and deploy deep learning models for image understanding on inspection systems.
• Advance KLA's inspection platforms with physics-based modeling or signal processing.
• Collaborate with cross-functional teams and customers; travel internationally as needed.

Skills

Deep learning / CV

Python

PyTorch / TensorFlow

C++ production software

Physics-based modeling

Publications

Education

PhD in Applied Math / CS / EE

MS in related field

BS in related field

Tools

C++

Python

Linux

Job description

KLA is seeking a motivated AI Engineer for its RAPID division in Milpitas, CA. You will design, train, and deploy deep learning models blended with physics-based methods to advance KLA's advanced reticle inspection platforms. This role requires collaboration across teams, dataset curation, and delivering production‑quality code.

International travel may be involved. Candidates should have a PhD (or 1–2 years) in Applied Mathematics, CS, EE or related field, strong CV/DL experience, proficiency