Artificial Intelligence Engineer - Milpitas, CA, United States
$124,100 - $211,000 USD yearly
Originally posted 29 July 2026 by the employer. Relisted by the employer on 20 October 2026.
About the role
This role focuses on designing, developing, modifying, and diagnosing complex machine learning and deep learning algorithms within KLA's LS-SWIFT Division. The position supports patterned wafer inspection systems used in high-volume semiconductor manufacturing.
What you'll do
- Design, develop, modify, and diagnose complex ML and DL algorithms. - Invent, prototype, and deploy new image processing algorithms and ideas. - Apply hands-on experience in object-oriented and functional programming. - Collaborate across engineering teams and operate effectively in an independent environment.
What you'll need
- Master's degree with 0 years of related work experience, or Bachelor's degree with 2 years of related work experience. - Strong background in machine learning, deep learning, computer vision, or image processing. - Familiarity with deep learning frameworks such as TensorFlow or PyTorch. - Hands-on experience training ML/AI models that have been deployed in a product. - Ability to write model architectures from scratch and modify loss functions. - Hands-on experience in object-oriented and functional programming.
Skills: ML/DL algorithms, image processing, TensorFlow, PyTorch, deep learning frameworks, model architectures
This role has been open 83 days — well above the 47-day median for Software Engineering Semiconductor roles.
Software Engineering Semiconductor · AI EDA Design
|
Open roles in category
106
|
Median days open
47 d
|
Median salary
$210k
|
See the full market breakdown ▾Category comparison, skills in demand, and who else is hiring
| Metric | KLA Corporation | All employers we track in this specialty (106 roles · 32 employers) |
|---|---|---|
| Open roles in this specialty | 14 | 106 |
| Open roles in the wider Software, Firmware & Systems family | 207 | 6222 · 145 employers |
| Median days open | 40 d | 47 d (−7 d vs this employer) |
| Median salary (USD postings) | — | $210k |
Skills observed across this category: ML/DL algorithms, image processing, TensorFlow, PyTorch, deep learning frameworks, model architectures
Who's hiring in this category
- Micron Technology · 14 open roles · median 39 d
- Applied Materials · 13 open roles · median 62 d
- KLA Corporation (this employer) · 13 open roles · median 32 d
- Lam Research · 8 open roles · median 24 d
- Cadence Design Systems · 7 open roles · median 22 d
- NVIDIA · 7 open roles · median 166 d
How we counted: 106 open Software Engineering Semiconductor (AI EDA Design) roles from 32 employers tracked in the SemiconductorJobs index, counted 20 Oct 2026. Specialty figures count only roles carrying this exact specialty label, so an employer's related work in neighbouring specialties is not included there — it is counted in the wider Software, Firmware & Systems family row. Figures refresh nightly.