Artificial Intelligence Engineer (Lead/Manager)

  • Chennai
  • Kla

Company Overview


KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Virtually every electronic device in the world is produced using our technologies. No laptop, smartphone, wearable device, voice-controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us. KLA invents systems and solutions for the manufacturing of wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. The innovative ideas and devices that are advancing humanity all begin with inspiration, research and development. KLA focuses more than average on innovation and we invest 15% of sales back into R&D. Our expert teams of physicists, engineers, data scientists and problem-solvers work together with the world’s leading technology providers to accelerate the delivery of tomorrow’s electronic devices. Life here is exciting and our teams thrive on tackling really hard problems. There is never a dull moment with us.


  • Design and develop machine learning software solutions for semiconductor device and thin film metrology
  • Work with both internal and external customers to define requirements for these software solutions
  • Build machine learning-based software solutions
  • Optimize algorithms and prototypical solutions for efficient implementation
  • Design, develop, implement, oversee, and adapt API
  • Extend prototypes into fully functional, polished solutions ready for internal and/or external use
  • Implement solutions in Python with TensorFlow
  • Write comprehensive unit tests and provide technical support during alpha and beta testing
  • Document the design and implementation of the software solutions, and give presentations both internally and externally regarding the work
  • Maintain new and existing software solutions throughout the life-cycle of our tools


Preferred Qualifications:

  • Strong background in machine learning using regression models and TensorFlow framework
  • Background in software architecture, modern development practices, and design patterns
  • Background in Python or C#, especially in writing production-quality, object-oriented code
  • Background in parallel / distributed computing, performance profiling, and memory optimization
  • Experience in API development is a plus
  • Experience in Azure DevOps is a plus
  • Experience in computational physics and numerical simulations is a plus
  • Experience in semiconductors (electrical properties, manufacturing, and metrology) is a plus
  • Ability and willingness to travel worldwide to support projects at customer sites is a plus




We offer a competitive, family friendly total rewards package. We design our programs to reflect our commitment to an inclusive environment, while ensuring we provide benefits that meet the diverse needs of our employees.


KLA is proud to be an equal opportunity employer