HPC / AI Software Infrastructure Lead (E)

To make electronics, you need chips, wafers, transistors, reticles, and... To make these, you must see, test and manufacture them at scale—faster and better than ever before. That's where KLA comes in. Whether you're early in your career or an experienced professional, you'll solve complex challenges, work alongside brilliant minds and help shape the future of technology.

What You'll Do

In this role, you will play a key part in advancing business priorities by delivering high-impact work across your area of expertise.

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

About KLA

We provide advanced inspection tools, metrology systems, process solutions, and computational analytics that make electronics possible, tackling complex challenges. From electron and photon optics to machine learning and data analytics, we seek perfection at the most fundamental level of matter in the universe. If you want to make electronics that push industries forward and make the world a better place, join us.

Total Rewards

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

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.

Use of AI Statement

At KLA, our interviews seek to understand your individual skills, problem-solving approach and authentic thinking. To ensure a fair and consistent evaluation, the use of AI, recording tools or other technologies to generate, suggest or provide responses during interviews—whether virtual or in person—is not permitted unless explicitly approved in advance as part of a reasonable accommodation or invited by the interviewer. Use of these tools may interfere with our ability to evaluate your individual qualifications and affect your candidacy. KLA is committed to advancing innovation through responsible AI, and we value candidates who share this mindset.

Equal Opportunity Statement

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 talent.acquisition@kla.com or at +1-408-352-2808 to request accommodation.

For additional information, view the US Know Your Rights poster on the U.S. Equal Employment Opportunity Commission website.