AI & HPC Infrastructure Lead

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

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.