AI HPC Engineer

AI HPC Engineer



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 in 2019 we invested 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.


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. Central Engineering is KLA’s largest engineering organization comprised of 9 Centers-of-Excellence (CoE) in various disciplines applied across all product groups in the company. These CoE include Handling & Automation, Precision Motion Control, Sensors & Image Acquisition, Platform Design, and Packaging Engineering, among others. Talent includes over 500 engineers across global centers in Israel, China, India, and the US. Each CoE contributes not just talent and deliverables per discipline toward product programs, but also subject matter expertise, best practices, roadmaps, specialized facilities, apparatus, models, and analytics. These differentiate KLA not only in WHAT we do, but also in HOW we do it.

About KLA Advanced Computing Labs, India

KLA advanced computing Labs’ (ACL) mission in India is to deliver advanced parallel computing research and software architectures for AI + HPC + Cloud solutions to accelerate KLA’s product performance. This team explores high-risk approaches, pioneering technologies, and novel methods to accelerate KLA’s algorithms and contribute to KLA’s HPC technology roadmap. We engage leading thinkers in academia, industry and KLA’s business units to create innovative parallel computing methods to enable KLA’s business growth.

New College Graduate Position: AI HPC Engineer

KLA is hiring engineers for its Advanced Computing Labs in Chennai, India. KLA ACL is at our new research center in the IITM, Research Park. The goal of the center is to conduct computational research and development in parallel and distributed sub-systems and deploy them to KLA’s advanced semi-conductor platforms that are used for inspection and metrology tasks in leading fabs. These efforts are part of a larger global initiative at KLA to scale up its AI + HPC + cloud infrastructure.

What will you be responsible for?

As part of this elite R&D team, the job entails broad understanding core algorithms that have to be expressed in various parallel computing constructs particularly HPC accelerators such as GPUs. The first step in optimizing will be to theoretically model break-down of our AI algorithms and model it in terms of available bandwidth, computational FLOPS etc. This is a technical hands-on role in exploration and implementation of these techniques and delivering production quality code. The implementation steps will include CUDA level programming along with performance tuning to ensure that we can come close to achieving the theoretical model. You will be exposed to a variety of image processing, signal processing and deep learning loads that have to be optimized. A complimentary stage of optimization includes exploring existing libraries and programming in higher level constructs such C++ Parallel programming.

You will collaborate with peer engineers in parallel computing areas and with algorithm teams in product groups.

What we would like to see?
New/recent College graduates in Ph.D. (preferred), MS in EE, CS or CSE or Math, Physics. Bachelors graduates will also be considered.
An engineer who has a strong foundation in computer architecture, and in particular with a special interest in high performance parallel processing at the device level (GPUs or CPUs/SIMD).
The engineer should have a strong mental model of computational loads and mapping different algorithms to parallel architectures.
Proficient in programming skills in C/C++/Python.
Good understanding and exposure to the Linux operating system at the user level.
Exposure to multiprocessor and multithreading concepts
The position also requires a person with significant communication, initiative and the ability to navigate from relatively high-level requirements to low level computational models.

Bonus Skills

Some familiarity with GPU programming such as CUDA, OpenCL or SYCL. Any prior exposure to programming and optimization in CUDA will be a very big plus. Additionally, any exposure to optimizing large scale signal or computer vision algorithms domain would also be a major plus.

Exposure to analyzing and tuning applications using profiling tools such as NSIGHT or VTUNE.

Exposure to large scale distributed HPC systems, experience in Docker and Container orchestration and any expertise in AI Frameworks (Tensorflow) is advantageous.

Finally a strong background in Modern C++ concepts (C++ 11 through C++ 17), STL library would also be a way to stand out from the crowd.

We offer a competitive, family friendly total rewards package. We design our programs to reflect our commitment to an inclusive environment, while ensuring we provide benefits that meet the diverse needs of our employees.

KLA is proud to be an equal opportunity employer

To apply for this job please visit

    XTechalpha: Launching a “next-in class“ platform of exponential technologies: AI, Blockchain, Cloud, Cybersecurity, IoT & emerging segments: GCCs, Startups and Academic Institutions & Edtech accentuated with an enviable line up of proven and seasoned thought leaders bringing together and ensemble, curated content, networking interventions and learning enhancement. A must visit primer for GIC heads, Technology leaders, Exponential technologies enthusiasts, Academic deans and Entrepreneurs.

    Watch the Launch video:

    XTechalpha: reimagine your THINKING