Kla
HPC / AI Software Infrastructure Lead (E)
Ann Arbor, MI · Mid
Sponsorship not specified$151k-$257kDetected 30 days ago
PythonC++Distributed SystemsAlgorithmsCI/CDLinuxDevOpsMachine LearningDeep LearningTensorFlowPyTorchData EngineeringComputer VisionAR/VRResearchLeadershipCommunicationMentoring
About the role
- Company Overview KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem.
- No laptop, smartphone, wearable device, voice-controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us.
Responsibilities
- 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
- Mentor and develop engineers, providing technical guidance, coaching, and growth opportunities for junior team members
- Proven success building distributed systems in HPC, AI/ML, or cloud-native environments
- Passion for developing talent and building inclusive, high-performing teams
- Collaborate across hardware, software, and algorithm teams in a deeply technical environment
Requirements
- 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
- 10+ years in software engineering, including leading and scaling technical teams
- Track record of delivering performance-critical infrastructure at scale
- Experience mentoring and growing early- and mid-career engineers
- Technical Expertise
- Experience with GPU computing (CUDA, ROCm) and modern AI frameworks (PyTorch, TensorFlow, etc.)
- Familiarity with high-performance storage systems, networking, and data pipelines
- Ability to operate as both a hands-on engineer and strategic technical leader
Nice to have
- Company Overview
- KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem.
- 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.
Skills
- Virtually every electronic device in the world is produced using our technologies.
Compensation
- $151,100.00 - $256,900.00
- 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.
Benefits
- Interns are eligible for some of the benefits listed.
- Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training.
Company info
- We are committed to complying with all applicable federal and state minimum wage requirements where applicable.
Equal opportunity
- KLA is proud to be an Equal Opportunity Employer.
- Please contact us at talent.acquisition@kla.com or at +1-408-352-2808 to request accommodation.
- Equal Opportunity Employer.
This listing is sourced directly from Kla's careers page and normalized into a canonical job model.