Software Engineer (AI/ML)
About The Position
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. 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.
Requirements
- 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.
- Master's Level Degree and 0 years related work experience; Bachelor's Level Degree and related work experience of 2 years
Nice To Haves
- 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.
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.
Benefits
- medical
- dental
- vision
- life
- 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
- paid company holidays
- family care and bonding leave
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Entry Level
