Senior AI Infrastructure Engineer

Job Description

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.

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

• Employees at KLA are often offered competitive pay with bonuses, a 401(k) match, an employee stock purchase program, and financial perks like student-debt assistance, planning support, and group insurance discounts. Health and lifestyle benefits typically include medical/dental/vision, life and other voluntary coverages, paid time off and holidays, family leave, backup care, wellness rewards, gym discounts, and community-volunteering opportunities. Employees also get strong growth support through tuition reimbursement, KLA’s corporate learning center, education awards, and engineering certification programs.