> Markdown version of [/jobs/ext/1701294-ai-engineer](https://www.wearedevelopers.com/jobs/ext/1701294-ai-engineer). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# AI Engineer

- **Company:** KLA Corporation
- **Location:** Milpitas, CA, United States
- **Experience:** Expert
- **Salary:** $136,300.0 - $231,700.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Computer Vision, C++ (Programming Language), Imaging Technology, Python (Programming Language), Object-Oriented Software Development, Tensorflow, Computational Geometry, Scientific Computating, Signal Processing, Pytorch, Deep Learning, Information Technology
- **Published:** July 7, 2026
- **Apply:** https://www.dice.com/job-detail/c9e19e96-2530-49ef-86ce-d774c23b5839

## About the Role

KLA is seeking a motivated AI Engineer with a growth mindset to join a world-class algorithm group within the Reticle And Photomask Inspection Division (RAPID)! In this role, you will design and develop deep learning models blended with classical methods (e.g., physics-based modeling or signal processing) to advance KLA's innovative inspection platforms., * PhD or MS in Applied Mathematics, Computer Science, Electrical Engineering, or a related field

 * Significant experience in deep learning, especially for computer vision and image understanding
 * Proficiency with modern deep learning frameworks (e.g., PyTorch or TensorFlow) and Python
 * Experience in C++ production software development and object-oriented programming

Desired Expertise

 * Scientific computing, signal processing, numerical optimization, and computational geometry
 * Modeling of imaging systems (electron-beam or optical)
 * Knowledge of semiconductor manufacturing and process control is a plus, Doctorate (Academic) Degree and 0 years related work experience; Master's Level Degree and related work experience of 3 years; Bachelor's Level Degree and related work experience of 5 years

## Description

Deep Learning & Computer Vision

 * Design, train, and deploy deep learning models for image understanding tasks such as image translation, segmentation, anomaly detection, and defect analysis on inspection systems
 * 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

Innovation and Execution

 * 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
 * Improve model accuracy; optimize models for inference throughput, including GPU-accelerated and high-performance computing

Cross-functional Collaboration

 * 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 at customer sites, including occasional international travel


## Related Videos

- [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning)
- [DevOps for AI: running LLMs in production with Kubernetes and KubeFlow](https://www.wearedevelopers.com/videos/1222-devops-for-ai-running-llms-in-production-with-kubernetes-and-kubeflow)
- [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs)
- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [Developing an AI.SDK](https://www.wearedevelopers.com/videos/198-developing-an-ai-sdk)
- [Geometric deep learning for drug discovery](https://www.wearedevelopers.com/videos/264-geometric-deep-learning-for-drug-discovery)

## Related Articles

- [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)
- [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud)
- [What Industries Outside of AI Are Hiring The Most AI Experts?](https://www.wearedevelopers.com/magazine/98-what-industries-outside-of-ai-are-hiring-the-most-ai-experts)
- [Stephan Gillich - Bringing AI Everywhere](https://www.wearedevelopers.com/magazine/489-stephan-gillich-bringing-ai-everywhere)
- [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development)
- [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)