Senior AI Infrastructure 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 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

Lead the Future of AI in Semiconductor Manufacturing

Join us in revolutionizing semiconductor manufacturing by developing modern AI models that detect nanoscale defects—often with limited data and system access. This is your opportunity to shape the future of AI in the semiconductor industry!

We’re looking for a visionary leader who thrives with solving hard problems, mentoring others, and delivering real-world impact. From architecture to deployment, you’ll be at the forefront of innovation, driving the development of intelligent systems that power the next generation of semiconductor technology!

Job Duties, but not limited to:

• Architect and build the MLOps backbone of our AI products, ensuring scalability, reliability, and performance.
• Collaborate multi-functionally with team members, designers, and engineers to define and implement technical requirements.
• Deploy sophisticated ML models at customer sites, ensuring seamless integration and high performance in production air-gapped environments.
• Stay ahead of the curve by integrating the latest ML/AI research into real-world applications.
• Lead the data pipeline—from dataset selection and preprocessing to quality assurance and model training.
• Lead experimentation and statistical analysis to fine-tune models for optimal accuracy and efficiency.
• Mentor and inspire junior engineers, fostering a culture of technical excellence and continuous learning.

Qualifications:

• A degree in Computer Science, Mathematics, or a related field.
• 7+ years of hands-on experience developing and deploying ML models for external customers—not just internal tools.
• Deep expertise in Python and ML frameworks like PyTorch, TensorFlow, and Keras.
• Proven experience with foundational models and modern AI toolchains.
• Strong analytical skills and a track record of solving sophisticated technical problems.
• Excellent communication skills—you can explain sophisticated concepts to both technical and non-technical audiences.
• Demonstrated leadership in managing projects or teams, with the ability to juggle multiple priorities in a fast-paced environment.

Why Join Us?

This is more than a job—it’s a chance to lead transformative work in a field that’s shaping the future of technology. You’ll work with forward-thinking minds, modern tools, and global customers who are pushing the boundaries of what’s possible.

Minimum Qualifications

Master's Level Degree and 3 years related work experience; Bachelor's Level Degree and related work experience of 5 years

Base Pay Range: $100,000.00 - $170,000.00 Annually

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