Senior AI Engineer

KLA; Milpitas (CA); On-site; USD 159,500 - 271,200; Full time

Benefits offered by this job

Medical, dental, and vision benefits

401(k) with company matching

Tuition reimbursement

Paid time off

Job summary

A leading global technology company in Milpitas is seeking a Senior AI Engineer to develop and implement agentic AI solutions. The ideal candidate will have over 5 years of experience in AI/ML development, particularly with agent-based systems and reinforcement learning. Responsibilities include working on multi-agent systems and collaborating with cross-functional teams to enhance semiconductor manufacturing processes. This position offers a competitive salary and a comprehensive benefits package.

Qualifications

• 5+ years in AI/ML development, exposure to agent-based systems.
• Experience with LLM pre-training is a significant advantage.
• Total post-graduation experience of 3-8 years required based on degree.

Responsibilities

• Architect and implement agentic AI solutions for semiconductor applications.
• Develop multi-agent systems that collaborate across semiconductor domains.
• Integrate reasoning, planning, and reinforcement learning into AI workflows.
• Collaborate to ensure seamless deployment and usability.

Skills

AI/ML development

Agent-based systems

Reinforcement learning

Python

JavaScript

Angular/Vue

C/C++

PyTorch

Web-based front-ends

Communication skills

Education

Master’s degree or PhD in Computer Science or Electrical Engineering

Job description

Company Overview

KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Our technologies enable virtually every electronic device in the world. KLA invents systems and solutions for wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. We focus on innovation and invest 15% of sales back into R&D. Our teams include physicists, engineers, data scientists and problem-solvers who tackle hard problems to accelerate tomorrow’s electronic devices.

Group/Division

Enabling the movement toward advanced chip design, KLA's Measurement, Analytics and Control group (MACH) is seeking research scientists, software engineers, application development engineers and senior product technology process engineers to join our team. MACH collaborates with customers to detect and control highly complex process variations at the source, rather than compensating later. With over 40 years of semiconductor process control experience, MACH provides patterning process analytics and control technologies, modeling, data analytics and process control modeling. As a member of the MACH team, you’ll join a leading process-control company working across functions to solve complex technical problems.

Job Description / Preferred Qualifications

The software data analytics division is looking for a Senior AI Engineer with experience developing, evaluating, deploying, and monitoring Agentic AI applications that go beyond traditional machine and deep learning.

In this role, you will work with the division’s AI Working Group to integrate Agentic AI across products to solve complex challenges in semiconductor manufacturing. You will participate in technical/architectural design, identify suitable technology stacks, and hands-on development, collaborating with a cross-functional team to integrate intelligent agents into applications and workflows that enhance efficiency, quality and innovation.

Responsibilities

• Architect and implement agentic AI solutions for semiconductor applications.
• Develop multi-agent systems that collaborate across semiconductor domains.
• Integrate reasoning, planning, and reinforcement learning into AI workflows.
• Collaborate with hardware, software and data teams to ensure seamless deployment, usability and user experience.
• Continuously evaluate emerging technologies in agent-based AI and apply them to real-world semiconductor challenges.

Skills & Experience

• 5+ years in AI/ML development, with strong exposure to agent-based systems or reinforcement learning, having designed, evaluated, deployed, and monitored such applications.
• Experience with LLM pre-training is optional, but a significant advantage.
• At least 5 years of total post-graduation working experience in software engineering, preferably including Machine Learning and Deep Learning.
• Proficiency in Python, JavaScript, Angular/Vue, C/C++, PyTorch.
• Experience developing web-based front-ends.
• Understanding of semiconductor manufacturing processes and data pipelines is a plus.
• Understanding of cloud platforms and MLOps for scalable AI deployment.
• Strong communication skills and ability to work in a dynamic cross-functional environment.
• Flexibility with evolving requirements and onboarding latest technologies and models.

Academic Background

• Master’s degree or PhD in Computer Science, Electrical Engineering or related field.
• Minor in Mathematics is optional, but preferred.

Minimum Qualifications

Doctorate (Academic) Degree with related work experience of 3 years; Master’s Degree with related work experience of 6 years; Bachelor’s Degree with related work experience of 8 years.

Base Pay Range: $159,500.00 - $271,200.00 Annually. Primary Location: USA-CA-Milpitas-KLA. KLA’s total rewards package may include performance incentives and benefits including medical, dental, vision, life, 401(k) with company matching, ESPP, student debt assistance, tuition reimbursement, development and career growth programs, financial planning, wellness benefits, paid time off, holidays and family care leave.

Interns are eligible for some benefits listed. Pay ranges are determined by role, level and location. Actual pay depends on factors including state minimums, location, skills, experience, and education. We comply with applicable federal and state minimum wage requirements where applicable. Your recruiter can share more about the specific pay range for your location during the hiring process.

Equal Opportunity and Accommodation

KLA is an Equal Opportunity Employer. We provide reasonable accommodation to qualified individuals with disabilities for the job application or interview process, to perform essential job functions, and to receive benefits of employment. To request accommodation, contact [e-mail] or [phone].

Be aware of potentially fraudulent job postings or recruiting activity. KLA never asks for financial compensation to be considered for an interview or employment, and does not work with recruiters who charge fees. Please ensure you review legitimate job postings on KLA’s Careers website. If you have concerns about the legitimacy of a communication or interview, email [e-mail] to confirm the person you are communicating with is an employee. We take your privacy seriously and handle information confidentially.