KLA Careers

AI Software Engineer

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 data.  Our optical and electron beam inspection and measurement tools use sophisticated data analytic techniques to provide value for our customers.
In this role, you will be part of a world class team of data science researchers, machine learning and application engineers who build and use a world class platform for development of Industrial AI application.
We are looking for engineers with a passion for data science and software engineering who can rapidly incorporate evolving ideas from academia and industry into useful solutions for our internal and external customers.


Engineers in the Central AI Software team will be working on building and maintaining infrastructure necessary for large scale experimentation and deployment of machine learning based solutions.  Domains in which a successful candidate will be expected to contribute will include data management and data loading, support for machine learning and deep learning model training, experimentation and deployment.

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
  • MPI and similar distributed computing framework.
  • GPU Architectures and CUDA (CuGraph, CuData, CuML etc).
  • Distribution frameworks like DASK.
  • GO Systems Programming, Java Programming.
  • Traditional machine learning using Random Forest, XG Boost, Logistic Regression.
  • Deep Learning for regression or classification.  Generative models.
  • TensorFlow, NumPy, scikit-learn, and other ML and DL frameworks.

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 and motivated by team success
  • Strong problem solving skills

Minimum Qualifications

PhD with at least 2 years OR
Master's Level Degree with at least 3 years of experience OR
Bachelor's Level Degree with at least 5 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