Software Engineer (AI/ML)

Position OverviewKLA is seeking a highly motivated Software AI Engineer to develop next-generation artificial intelligence and machine learning solutions supporting semiconductor process control, inspection, and metrology systems. The successful candidate will work at the intersection of AI, computer vision, algorithm development, and high-performance software engineering to deliver scalable solutions that drive product innovation and customer value.Key ResponsibilitiesDesign, develop, and deploy AI/ML algorithms for semiconductor inspection, metrology, and process control applications.Build and optimize machine learning models using large-scale engineering and manufacturing datasets.Develop software systems supporting model training, validation, deployment, and inference workflows.Collaborate with algorithm engineers, software engineers, application engineers, and product teams to integrate AI capabilities into production systems.Evaluate emerging AI technologies and develop proof-of-concept solutions for new product capabilities.Improve performance and scalability through optimization techniques including parallel computing, GPU acceleration, and efficient software architectures.Contribute to cloud-enabled and edge-computing AI solutions supporting semiconductor equipment platforms.Drive technical innovation while maintaining software quality, reliability, and maintainability.Required QualificationsMS in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, Artificial Intelligence, Machine Learning, or a related field.Strong programming experience in Python and C++.Experience developing and deploying machine learning or deep learning solutions.Solid understanding of data structures, algorithms, software engineering principles, and object-oriented design.Experience working in Linux development environments.Strong analytical and problem-solving skills.Preferred QualificationsExperience with computer vision, image processing, computational imaging, or machine vision systems.Experience with deep learning frameworks such as TensorFlow or PyTorch.GPU programming experience using CUDA or other acceleration technologies.Experience with semiconductor manufacturing, metrology, inspection, or process control applications.Exposure to MLOps, cloud computing, distributed systems, or large-scale data platforms.Experience productizing AI solutions in a commercial software environment.Minimum QualificationsMaster's Level Degree and 0 years related work experience; Bachelor's Level Degree and related work experience of 2 yearsBase Pay Range: $97,600.00 - $165,900.00 AnnuallyKLA'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 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.KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [email protected] or at +1-408-352-2808 to request accommodation.Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees. KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA . Please ensure that you have searched KLA's Careers website for legitimate job postings. KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers. If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to [email protected] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.