HPC / AI Software Infrastructure Lead (E)

KLA; Ann Arbor Charter Township (MI); On-site; USD 151,100 - 256,900; Full time

Benefits offered by this job

Medical benefits

401(K) with company matching

Tuition reimbursement program

Paid time off

Job summary

KLA in Ann Arbor is looking for an HPC/AI Software Infrastructure Lead to architect and scale advanced AI and HPC systems. You will lead teams to design high-performance distributed systems and establish engineering best practices.

The role requires 10+ years of software engineering experience and expertise in C++, Python, and distributed systems. KLA offers comprehensive benefits and a commitment to innovation.

Qualifications

• 10+ years in software engineering, including leading and scaling technical teams.
• Proven success building distributed systems in HPC, AI/ML, or cloud-native environments.
• Experience mentoring and growing early- and mid-career engineers.

Responsibilities

• Lead the architecture and development of large-scale HPC and AI infrastructure.
• Design scalable distributed systems unifying traditional image processing with AI.
• Establish engineering best practices for mission-critical systems.

Skills

Leadership

C++

Python

Distributed Systems

GPU Computing

AI/ML

Education

Doctorate Degree and related work experience

Master’s Level Degree and related work experience

Bachelor’s Level Degree and related work experience

Tools

CUDA

PyTorch

TensorFlow

Job description

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 we invest 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.

Job Description / Preferred Qualifications

HPC/AI Software Infrastructure Leads are core to KLA’s technology, while we do not currently have an opening, we are always building our HPC/AI Software Infrastructure Lead engineering talent community, we are interested in learning about your background.

Apply to this posting for Future Opportunities with KLA.

At KLA, we’re pushing the boundaries of semiconductor inspection through advanced AI and high‑performance computing. We are looking for a hands‑on technical leader to architect and scale the next generation of AI/HPC infrastructure powering our most critical imaging and data platforms. This role is ideal for someone who thrives at the intersection of distributed systems, GPU computing, and real‑world AI workloads, and who enjoys building and mentoring high‑performing engineering teams while driving technical excellence.

What You’ll Do

• Lead the architecture and development of large‑scale HPC and AI infrastructure supporting cutting‑edge image processing and machine learning workloads
• Design scalable, high‑performance distributed systems that unify traditional image processing with modern AI/Deep Learning pipelines
• Drive GPU‑accelerated computing strategies, optimizing performance across compute, storage, and networking layers
• Partner cross‑functionally with hardware, algorithms, and product teams to deliver robust, production‑ready platforms
• Establish engineering best practices (code quality, CI/CD, observability, performance tuning) for mission‑critical systems
• Mentor and develop engineers, providing technical guidance, coaching, and growth opportunities for junior team members
• Serve as a technical leader and decision‑maker, influencing architecture and long‑term platform strategy

What You Bring

Experience

• 10+ years in software engineering, including leading and scaling technical teams
• Proven success building distributed systems in HPC, AI/ML, or cloud‑native environments
• Track record of delivering performance‑critical infrastructure at scale
• Experience mentoring and growing early‑ and mid‑career engineers

Technical Expertise

• Deep understanding of distributed systems, parallel computing, and Linux systems programming
• Strong programming skills in C++, Python, or similar systems‑level languages
• Experience with GPU computing (CUDA, ROCm) and modern AI frameworks (PyTorch, TensorFlow, etc.)
• Familiarity with high‑performance storage systems, networking, and data pipelines
• Strong foundation in CI/CD, DevOps, and production system reliability

Bonus Experience

• Background in image processing, computer vision, or scientific computing
• Experience supporting hybrid HPC + AI workloads in production environments

Leadership & Impact

• Passion for developing talent and building inclusive, high‑performing teams
• Ability to operate as both a hands‑on engineer and strategic technical leader
• Strong communication skills with the ability to influence across engineering and product stakeholders

Why KLA / Why Ann Arbor

• Work on real‑world AI systems at scale, not just experiments
• Collaborate across hardware, software, and algorithm teams in a deeply technical environment
• Join a growing engineering presence in Ann Arbor, with access to top talent and a strong technical community
• Opportunity to shape the direction of AI infrastructure in a core product domain

Minimum Qualifications

• Doctorate (Academic) Degree and related work experience of 5 years
• Master’s Level Degree and related work experience of 8 years
• Bachelor’s Level Degree and related work experience of 12 years

Base Pay Range: $151,100.00 – $256,900.00
Primary Location: USA-MI-Ann Arbor

KLA’s total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to medical, dental, vision, life, and other voluntary benefits, 401(K) including company matching, employee stock purchase program (ESPP), student debt assistance, tuition reimbursement program, development and career growth opportunities and programs, financial planning benefits, wellness benefits including an employee assistance program (EAP), paid time off and paid company holidays, and family care and bonding leave. Interns are eligible for some of the benefits listed. Our pay ranges are determined by role, level, and location. The range displayed reflects the pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including state minimum pay wage rates, location, job‑related skills, experience, and relevant education level or training. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [e-mail] or at [phone] to request accommodation.

Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees. KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA’s Careers website for legitimate job postings. KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers. If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to [e-mail] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.