KLA Careers

AI Software Engineer

This job posting is no longer active

Company Overview

Calling the adventurers ready to join a company that's pushing the limits of nanotechnology to keep the digital revolution rolling. At KLA, we're making technology advancements that are bigger—and tinier—than the world has ever seen.

Who are we?  We research, develop, and manufacture the world's most advanced inspection and measurement equipment for the semiconductor and nanoelectronics industries. We enable the digital age by pushing the boundaries of technology, creating tools capable of finding defects smaller than a wavelength of visible light. We create smarter processes so that technology leaders can manufacture high-performance chips—the kind in that phone in your pocket, the tablet on your desk and nearly every electronic device you own—faster and better. We're passionate about creating solutions that drive progress and help people do what wouldn't be possible without us.  The future is calling. Will you answer?


KLA has always had a close relationship with physics and data.  Our optical and electron beam inspection and measurement tools use cutting edge physics models, both for hardware design and as part of their algorithms.  AI, including several traditional machine learning techniques and deep learning are routinely used to process this data to meet application requirements.

The AI & Modeling Center of Excellence, centered in KLA’s R&D facility in Ann Arbor, MI, was setup with the mission of advancing KLA’s traditional strengths in physics and data and providing implementation solutions for multiple KLA Inspection and Metrology products targeted at the semiconductor manufacturing industry.


As a part of this group, you will be part of a world class team of physicists, HPC system designers, machine learning and application engineers who build cutting edge solutions for modeling complex imaging techniques and semiconductor processes.  You will also work with a data scientists and AI infrastructure engineers whose mission is to build and scale machine learning based solutions for our semiconductor customers.

We are looking for engineers in a few different fields.  If you are passionate about Physics Modeling, High Performance Computing - HPC (including GPU), ML, Data, or Cloud technologies – this is the place for you!


Software Engineers in the AI and Physics Modeling team may work on a variety of tasks including platform for large scale experimentation, scaleout of physics models, data management and inference solutions for KLA products. 

Although familiarity with Machine Learning and Deep Learning solutions would be a big plus, this is primarily a Software Engineering position.  Successful candidates are passionate about software, and will have exceptional skills and hands on experience with development in C/C++ and Python in a Unix environment.  Deep conceptual understanding of multi threaded and multi process software systems is also necessary.

In addition to the above fundamental software skills, any of the following technical hands-on skills are highly desirable.

·      System Programming and Operating Systems.

·      Distributed systems for data management.

·      Distributed computing infrastructure.

·      Cloud technologies for storage, containerization and compute clusters.

·      Data Structures and algorithms

·      GPU architectures and data management.

·      Traditional machine learning using Random Forest, XG Boost, Logistic Regression.

·      Deep Learning for regression, classification.  Generative models.

·      TensorFlow, NumPy, scikit-learn, and other ML and DL frameworks.

·      MPI and similar distributed computing framework.

·      GPU Architectures and CUDA (CuGraph, CuData, CuML etc).

·      GO Systems Programming, Java Programming.

Successful candidates for this position will also demonstrate the following non-technical skills.

·      Capability to formulate creative solutions through analyzing complex data

·      Good communication skills

  • Strong team player

·      Strong problem solving skills


Bachelor's Level Degree

Must have flexibility to travel outside of Michigan based on business needs.

Minimum Qualifications

Master's Level DegreeORBachelor's Level Degree with at least 1 years of experience.

Equal Employment Opportunity

KLA is an Equal Opportunity Employer. Applicants will be considered for employment without regard to age, race, color, religion, sex, sexual orientation, gender identity, national origin, protected veteran status, disability, or any other characteristics protected by applicable law.

Related Careers