Senior AI Infrastructure Engineer

KLA • Ann Arbor, MI

16d

About The Position

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!

Requirements

• 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.

Responsibilities

• 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.

Benefits

• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Number of Employees: 5,001-10,000 employees