AI Engineering Architect - Agentic AI & Digital Engineering

KLA; Milpitas (CA); On-site; USD 186,200 - 316,500; Full time

Benefits offered by this job

Medical, dental, vision, and life insurance

401(K) company matching

Employee stock purchase program

Paid time off and holidays

Tuition reimbursement program

Job summary

A leading semiconductor company in Milpitas is seeking an experienced engineer to own the AI roadmap for digital engineering use cases. The role involves building AI solutions for predictive monitoring and developing workflows for engineering task automation. Ideal candidates will have a PhD or Master's degree in relevant fields and extensive hands-on experience with AI/ML solutions. The position offers an attractive compensation ranging from $186,200 to $316,500 annually along with comprehensive benefits.

Qualifications

• Doctorate (Academic) Degree and related work experience of 5 years.
• Master's Level Degree and related work experience of 8 years.
• Bachelor's Level Degree and related work experience of 12 years.
• Strong hands-on experience building and deploying AI/ML solutions.

Responsibilities

• Own the AI roadmap for EUV source and digital engineering use cases.
• Build AI solutions for source health analytics and predictive monitoring.
• Develop AI-assisted workflows for root-cause analysis and task automation.
• Partner with teams to enable AI-assisted modeling and knowledge-grounded workflows.
• Support AI-assisted routing and early BOM activities.

Skills

Experience with agentic AI

Time-series forecasting

Anomaly detection

Model-based systems engineering (MBSE)

Design integration

Education

Master's or PhD in Computer Science, AI/ML, Data Science, Electrical Engineering, Mechanical Engineering, Physics, or Systems Engineering

Job description

Company Overview

KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Virtually every electronic device in the world is produced using our technologies. No laptop, smartphone, wearable device, voice‑controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us. KLA invents systems and solutions for the manufacturing of wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. The innovative ideas and devices that are advancing humanity all begin with inspiration, research and development. KLA focuses more than average on innovation and we invest 15% of sales back into R&D. Our expert teams of physicists, engineers, data scientists and problem‑solvers work together with the world's leading technology providers to accelerate the delivery of tomorrow's electronic devices. Life here is exciting and our teams thrive on tackling really hard problems. There is never a dull moment with us.

Group/Division

With over 40 years of semiconductor process control experience, chipmakers around the globe rely on KLA to ensure that their fabs ramp next‑generation devices to volume production quickly and cost‑effectively. Enabling the movement towards advanced chip design, KLA's Global Products Group (GPG), which is responsible for creating all of KLA's metrology and inspection products, is looking for the best and the brightest research scientist, software engineers, application development engineers, and senior product technology process engineers. The RAPID division is the world leading provider of reticle inspection solutions for the semiconductor industry. The company provides inspection solutions to both the mask shops and the semiconductor fabs to ensure that lithography yields are consistently high thus enabling cost-effective manufacturing.

Job Description / Preferred Qualifications

Key Responsibilities

• Own the AI roadmap for EUV source and digital engineering use cases; identify, prioritize, and sequence high-value opportunities tied to engineering impact.
• Build AI solutions for source health analytics, anomaly detection, and predictive monitoring using engineering telemetry, metrology, and operational data.
• Develop AI-assisted and agentic workflows for alarm audit, root‑cause analysis, procedure creation, and engineering task automation.
• Partner with engineering teams to enable AI-assisted modeling, digital thread, MBSE, and knowledge-grounded engineering workflows.
• Support AI-assisted routing, CAD integration, model-to‑BOM, and early BOM activities as part of engineering digital transformation.
• Define AI observability, evaluation, tracing, and safe production deployment practices for engineering AI systems.

Preferred Qualifications

• Experience with agentic AI, tool‑using LLM workflows, multi‑agent systems, or RAG/vector-backed knowledge systems.
• Experience with time‑series forecasting, anomaly detection, or predictive analytics for industrial systems.
• Experience with MBSE, digital thread, SysML, requirements traceability, or model-centric engineering.
• Experience with CAD/PLM/PDM integration, BOM workflows, or engineering data interoperability.
• Experience in semiconductor equipment, capital equipment, optics, plasma, vacuum systems, or manufacturing engineering.

Minimum Qualifications

Doctorate (Academic) Degree and related work experience of 5 years; Master's Level Degree and related work experience of 8 years; Bachelor's Level Degree and related work experience of 12 years

• Master's degree or PhD in Computer Science, AI/ML, Data Science, Electrical Engineering, Mechanical Engineering, Physics, Systems Engineering, or a related field.
• Strong hands‑on experience building and deploying AI/ML solutions for complex engineering or industrial workflows.
• Experience with integrating AI into engineering data, software, or workflow systems.
• Experience with engineering data such as sensor streams, logs, structured/unstructured technical documents, or time‑series datasets.
• Ability to translate ambiguous engineering pain points into scalable AI architectures and deployable solutions.

Base Pay Range: $186,200.00 - $316,500.00 Annually. Primary Location: USA-CA-Milpitas. 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 (EAP), paid time off and paid company holidays, and family care and bonding leave. Interns are eligible for some of the benefits listed. Our pay ranges are determined by role, level, and location. The range displayed reflects the pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [e-mail] or at [phone] to request accommodation.

Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees. KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA's Careers website for legitimate job postings. KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers. If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to [e-mail] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.