Software Engineer (AI/ML)

About this role

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.

Group/Division Enabling the movement toward advanced chip design, KLA's Measurement, Analytics and Control group (MACH) is looking for the best and brightest research scientists, software engineers, application development engineers and senior product technology process engineers to join our team. The MACH team's mission is to collaborate with our customers to innovate technologies and solutions that detect and control highly complex process variations—at their source—rather than compensate for them at later stages of the manufacturing process. With over 40 years of semiconductor process control experience, chipmakers around the globe rely on KLA to ensure that their fabs ramp next-generation devices to volume production quickly and cost-effectively. Our MACH team develops leading-edge solutions for patterning process analytics and control technologies, thereby providing customers with critical insight at the feature level, field level and cross-wafer analysis. Our teams also develop advanced modeling simulation, data analytics and process control modeling technologies. As a member of the MACH team, you’ll be joining the most sophisticated and successful process-control company in the semiconductor industry--working across functions to solve the most complex technical problems in the digital age.

Job Description/Preferred Qualifications Position Overview KLA 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 Responsibilities • Design, 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 Qualifications • MS 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 Qualifications • Experience 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 Qualifications Master's Level Degree and 0 years related work experience; Bachelor's Level Degree and related work experience of 2 years

Base Pay Range: $97,600.00 - $165,900.00 Annually

Primary Location: USA-TX-Austin-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 on several factors, includin

Similar jobs

• Software Engineer (AI/ML) — Mia-labs · Austin Office
• Software Engineer AI/ML — GE Aerospace · Remote
• Senior Software Engineer (AI/ML) — 540 · Patrick SFB, FL or Arlington, VA
• Software Engineer (AI) — Numeral · New York (hybrid)
• Software Engineer - AI — Promise · Washington, D.C.
• Software Engineer - AI — Brellium · New York City
• Software Engineer - AI — Promise · San Francisco
• Software Engineer - AI — Lightspeeddms · South Jordan, UT