KLA logo

Senior AI Engineer

KLA
April 26, 2026
On-site
Milpitas, California, United States
$136,300 - $231,700 USD yearly
Level - Senior

Job Title

Senior AI Engineer

Role Summary

Design and implement agentic AI solutions and integrate intelligent agents into products and workflows to address semiconductor manufacturing challenges. Work within the Measurement, Analytics and Control (MACH) group on technical architecture, hands-on development, and cross-functional deployment to improve efficiency, quality, and automation.

Collaborate with software, hardware, and data teams to embed agent-based workflows into production applications and scale via cloud and MLOps.

Experience Level

Senior-level. Minimum qualifications indicate: Doctorate may qualify with 0 years of experience; Master’s with ~2 years; Bachelor’s with ~3 years of relevant experience.

Responsibilities

Core responsibilities include technical design, development, and deployment of agentic AI systems:

  • Architect and implement agentic AI solutions for semiconductor applications.
  • Design and develop multi-agent systems that collaborate across domains.
  • Integrate reasoning, planning, and reinforcement learning into AI workflows.
  • Identify suitable technology stacks and lead hands-on implementation.
  • Collaborate with hardware, software, and data teams to ensure deployment, usability, and user experience.
  • Continuously evaluate and apply emerging agent-based AI technologies to production problems.
  • Develop monitoring and maintenance strategies for deployed AI agents.

Requirements

Must-have technical skills and experience:

  • Proven AI/ML development experience, including work on agent-based systems or reinforcement learning.
  • Software engineering experience with machine learning and deep learning workflows.
  • Proficiency in Python and C/C++; experience with PyTorch.
  • Experience with cloud platforms and MLOps for scalable deployment.
  • Experience developing web-based front-ends.
  • Strong communication skills and ability to work in cross-functional teams and adapt to evolving requirements.

Nice-to-have:

  • Experience with LLM pre-training.
  • Familiarity with JavaScript and frameworks such as Angular or Vue.
  • Domain knowledge of semiconductor manufacturing processes and data pipelines.

Education Requirements

Doctorate (PhD) or Master’s degree in Computer Science, Electrical Engineering, or a related technical field preferred; Bachelor’s degree accepted with relevant experience. A minor in Mathematics is listed as preferred. The posting also allows equivalent practical work experience in lieu of degree (e.g., Master’s with ~2 years, Bachelor’s with ~3 years).


About the Company

Company: KLA

Headquarters: Chennai, India

KLA is a global leader in diversified electronics for the semiconductor manufacturing industry. The company enables the production of electronic devices by inventing systems and solutions for manufacturing integrated circuits, wafers, and displays. With over 40 years of experience, KLA invests heavily in innovation and R&D to support advanced chip design and manufacturing process optimization, collaborating with top technology providers to deliver future electronic devices.

KLA logo

Date Posted: 2026-04-26