Sr. Software Engineer - AI​/ML

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.

the Information Technology (IT) group at KLA is involved in every aspect of the global business. IT’s mission is to enable business growth and productivity by connecting people, process, and technology. It focuses not only on enhancing the technology that enables our business to thrive but also on how employees use and are empowered by technology. This integrated approach to customer service, creativity and technological excellence enables employee productivity, business analytics, and process excellence.

We are seeking a highly skilled Senior Software Engineer with specialized expertise in Machine Learning, Artificial Intelligence, and Data Science to join our innovative technology team. This role will be key in developing impactful AI solutions that drive business transformation and technological advancement. The ideal candidate will combine strong technical skills with business intelligence to translate complex requirements into practical, scalable applications that deliver measurable impact.

• Design, develop, and implement machine learning models and AI algorithms to address business challenges, including data preprocessing, model training, validation, and deployment.
• Architect and design scalable systems that can handle large volumes of data and complex AI/ML workloads, ensuring high performance and reliability.
• Implement AI/ML solutions that are robust, maintainable, and ready for production environments, incorporating security, compliance, and operational efficiency considerations.
• Stay abreast of the latest developments in AI,ML, and data science, conduct research, and explore new methodologies to keep solutions innovative.
• Work closely with multi‑functional teams including data engineers, product managers, software developers, and business partners to integrate advanced solutions into products and services.
• Apply critical thinking to solve complex problems using innovative data analysis, model architecture, system integration, and business strategy approaches.
• Oversee the deployment of AI/ML models into production, ensuring scalability, performance, and reliability; monitor model performance post‑deployment and iterate as necessary.
• Document processes, methodologies, and findings, and share knowledge through internal presentations, workshops, or by contributing to the knowledge base.

• Bachelor’s degree in Computer Science, Data Science, AI, or a related field;
  Master’s or PhD preferred.
• Minimum of eight (8) years of software engineering experience, with 3–5 years in AI/ML and proven track record of deploying solutions in a production environment.
• Proficiency in Python or similar languages, and experience with libraries such as Tensor Flow, Scikit‑learn, etc.; experience with Retrieval‑Augmented Generation (RAG) is essential.
• Strong background in statistics, machine learning algorithms, deep learning, neural networks, and system architecture.
• Strong understanding of cloud platforms, Dev Ops practices, and data processing technologies.
• Demonstrated ability to lead technical projects and work…