AI Engineer (Computer Vision/Signal Processing)

About the Role

KLA is seeking a motivated AI Engineer to join the Reticle And Photomask Inspection Division (RAPID). This role focuses on designing and developing deep learning models integrated with classical methods to enhance KLA's inspection platforms. The team is dedicated to innovation in semiconductor manufacturing, ensuring high-quality production through advanced inspection solutions. Candidates will collaborate with cross-functional teams and support model deployments at customer sites.

Requirements

• PhD in Applied Mathematics, Computer Science, Electrical Engineering, or a related field, or a Master's degree with 3 years of experience, or a Bachelor's degree with 5 years of experience.
• Significant experience in deep learning for computer vision and image understanding.
• Knowledge in physics-based modeling or signal processing.
• Proficiency with modern deep learning frameworks such as PyTorch or TensorFlow and Python.
• Experience in C++ production software development and object-oriented programming.

Responsibilities

• Design, train, and deploy deep learning models for image understanding tasks such as image translation, segmentation, anomaly detection, and defect analysis.
• Abstract specific tasks into generalized mathematical problems and devise customized models and algorithmic solutions.
• Curate and manage datasets; establish rigorous training, validation, and evaluation pipelines.
• Drive innovation by proposing novel approaches and delivering production-quality, maintainable code.
• Identify technical bottlenecks and balance model sophistication against computational complexity through comprehensive data analysis.
• Collaborate directly with other teams to co-optimize algorithms along with other sub-systems for best overall system design.
• Work with field teams and customers to support model deployments and ensure seamless integration of AI solutions to inspection systems.

Compensation

$136,300 - $231,700 per year

Benefits

• Competitive pay with bonuses
• 401(k) match
• Employee stock purchase program
• Student-debt assistance
• Financial planning support
• Group insurance discounts
• Medical/dental/vision insurance
• Life and other voluntary coverages
• Paid time off and holidays
• Family leave
• Backup care
• Wellness rewards
• Gym discounts
• Community-volunteering opportunities
• Tuition reimbursement
• Corporate learning center access
• Education awards
• Engineering certification programs

Location: Milpitas, CA • Type: Full-time • Posted: 8/25/2026