AI ML Engineer

Role Overview KLA is a globally recognized leader in diversified electronics and semiconductor manufacturing technologies. The AI ML Engineer role is based at IIT Madras Research Park in Chennai and is part of KLA Advanced Computing Labs, a team focused on delivering cutting-edge parallel computing research and software architectures for AI, HPC, and Cloud platforms. This is an exciting opportunity for a driven engineer to work alongside world-class physicists, data scientists, and technology innovators shaping the future of semiconductor and electronics manufacturing. Key Responsibilities - Design and develop AI and machine learning models and solutions to accelerate KLA product performance - Contribute to research and development of advanced parallel computing architectures for HPC and Cloud environments - Collaborate with academia, industry partners, and internal business units to pioneer novel computing methods - Support the execution of KLA's HPC technology roadmap through innovative algorithm development - Explore high-risk, high-reward approaches and emerging technologies to solve complex engineering challenges Required Qualifications - Doctorate degree with a minimum of 3 years of relevant work experience, OR - Master's degree with a minimum of 6 years of relevant work experience, OR - Bachelor's degree with a minimum of 8 years of relevant work experience - Strong background in AI, machine learning, or parallel computing - Experience with HPC, Cloud computing, or semiconductor-related software is highly preferred - Demonstrated ability to work in a research-oriented, problem-solving environment Why Join Us KLA invests over 15 percent of its annual sales revenue back into research and development, making it one of the most innovation-driven companies in the global technology sector. Joining KLA means working on problems that genuinely matter, with a team that thrives on tackling hard challenges in a collaborative and inclusive environment.

- Proficiency in AI and machine learning frameworks - Experience with HPC or parallel computing architectures - Strong programming skills relevant to AI/ML development - Doctorate, Master's, or Bachelor's degree in a relevant technical discipline - Minimum 3 to 8 years of experience depending on educational qualification - Ability to collaborate with cross-functional and academic teams - Research orientation with a problem-solving mindset

- Competitive and family-friendly total rewards package - Inclusive workplace environment - Benefits designed to meet diverse employee needs - Opportunity to work at IIT Madras Research Park - Engagement with leading academic and industry thinkers - High-impact work in global semiconductor and electronics innovation