AI Software Engineer (ML) – Semiconductor Process Control

KLA-Belgium; Austin (TX); On-site; USD 98,000 - 166,000; Full time

Benefits offered by this job

Medical, dental, vision

401(k) with company matching

Employee stock purchase program (ESPP)

Paid time off and holidays

Career growth opportunities

Job summary

KLA is seeking a Software AI Engineer to develop AI/ML solutions for semiconductor process control, inspection, and metrology systems in Austin, TX. You will work at the intersection of AI, computer vision, and high-performance software engineering to deliver scalable, production-ready features and customer value.

The role emphasizes collaboration across algorithm, software, and product teams, with opportunities to optimize performance, scale, and deploy on cloud and edge platforms.

Qualifications

• MS in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, AI/ML, or related field.
• Strong programming experience in Python and C++.
• Experience developing and deploying ML solutions.
• Solid understanding of data structures, algorithms, OO design, and software engineering principles.
• Experience working in Linux development environments.

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 datasets.
• Develop software systems for model training, validation, deployment, and inference workflows.
• Collaborate with algorithm, software, application, and product teams to integrate AI into production systems.
• Improve performance and scalability via parallel computing and GPU acceleration.

Skills

Python

C++

Education

MS in Computer Science or related field

Tools

TensorFlow

PyTorch

Linux

Job description

KLA is seeking a Software AI Engineer to develop AI/ML solutions for semiconductor process control, inspection, and metrology systems in Austin, TX. You will work at the intersection of AI, computer vision, and high-performance software engineering to deliver scalable, production-ready features and customer value.

The role emphasizes collaboration across algorithm, software, and product teams, with opportunities to optimize performance, scale, and deploy on cloud and edge platforms.