Overview
Join KLA's Applied AI Research organization as an AI Engineer, combining software engineering fundamentals with machine learning, statistical algorithms, deep learning and agentic AI to build next-generation intelligence platforms for semiconductor manufacturing and engineering.
What you'll do
- Investigate emerging AI techniques and evaluate their applicability to semiconductor manufacturing and engineering workflows.
- Conduct hands-on research in areas such as foundation models, agentic AI systems, scientific machine learning, computer vision, reinforcement learning, probabilistic modeling, knowledge representation, multi-agent systems and causal AI.
- Develop novel approaches for solving difficult engineering and manufacturing problems.
- Design rigorous experimental frameworks to validate hypotheses and quantify impact.
- Translate research findings into scalable product capabilities.
- Design and develop advanced algorithms using Bayesian inference, probabilistic graphical models, design of experiments, time-series analysis, optimization techniques, causal inference, anomaly detection, predictive modeling, multivariate statistics and graph analytics.
- Create explainable and scientifically sound models and evaluate statistical significance of experimental outcomes.
- Build decision-support and autonomous reasoning systems.
- Develop next-generation deep learning solutions for defect management, yield analytics, manufacturing optimization, image understanding, pattern recognition and engineering knowledge discovery.
- Work with and innovate on CNNs, vision transformers, multimodal models, graph neural networks, self-supervised learning, representation learning, foundation models and diffusion architectures.
- Develop intelligent autonomous systems capable of planning, reasoning, tool usage, workflow orchestration and long-horizon goal achievement.
- Build AI systems using LangGraph, Semantic Kernel, AutoGen, CrewAI, OpenAI agent frameworks and MCP (Model Context Protocol).
- Develop production-grade solutions in Python and modern C++ (C++17/20) with PyTorch, Ray, Spark, Kubernetes, Docker, vector databases and graph databases.
- Transform research prototypes into scalable and maintainable products.
- Lead technical reading groups, review and discuss recent research papers, and drive internal innovation initiatives.
- Mentor junior engineers on research methodology, promote first-principles thinking, and establish engineering rigor around benchmarking and evaluation.
What you'll need
- MS or PhD in computer science, machine learning, artificial intelligence, statistics, applied mathematics, electrical engineering or a related STEM discipline.
- Typically BS + 5 years, or MS + 3 years, or PhD of related experience.
- Minimum: Doctorate and 0 years related work experience; Master's and 3 years related work experience; or Bachelor's and 5 years related work experience.
- Demonstrated experience in building production AI systems while maintaining strong interest in research and innovation.
Details
- Location: Chennai, India.
- Individual contributor role.
Read the full description and apply on the company’s own careers page.