Software Engineer - AI/ML

To make electronics, you need chips, wafers, transistors, reticles, and... To make these, you must see, test and manufacture them at scale—faster and better than ever before. That's where KLA comes in. Whether you're early in your career or an experienced professional, you'll solve complex challenges, work alongside brilliant minds and help shape the future of technology.

Group/Division

KLA's IT group supports business growth and productivity by connecting people, process and technology around the world. We work to improve the technology that drives our business to thrive and focus on empowering employee use of technology. This integrated approach to customer service, creativity and technological excellence enhances employee productivity, business analytics and process excellence.

What You'll Do

In this role, you will play a key part in advancing business priorities by delivering high-impact work across your area of expertise.

We are seeking a highly skilled 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.

Key Responsibilities:

• AI/ML Solution Development: Design, develop, and implement machine learning models and AI algorithms to address business challenges. This includes data preprocessing, model training, validation, and deployment.

• Scalable Architecture Design: Architect and design scalable systems that can handle large volumes of data and complex AI/ML workloads, ensuring high performance and reliability.

• Production-Ready Solutions: Implement AI/ML solutions that are robust, maintainable, and ready for production environments. This includes considerations for security, compliance, and operational efficiency.

• Innovation and Research: Stay abreast of the latest developments in AI, ML, and data science. Conduct research to explore new methodologies or improve existing ones, ensuring our solutions remain innovative.

• Collaboration: Work closely with multi-functional teams including data engineers, product managers, software developers, and business partners to integrate advanced solutions into our products and services, ensuring alignment with business objectives.

• Problem Solving: Apply critical thinking to solve complex problems, often requiring innovative approaches to data analysis, model architecture, system integration, and business strategy.

• Deployment and Monitoring: Oversee the deployment of AI/ML models into production environments, ensuring scalability, performance, and reliability. Monitor model performance post-deployment and iterate as necessary.

• Documentation and Knowledge Sharing: Document processes, methodologies, and findings. Share knowledge through internal presentations, workshops, or by contributing to our knowledge base.

Preferred:

• Experience with specialized AI domains or industry-specific applications.

• Contributions to open-source projects or research publications in AI/ML.

• Knowledge of AI ethics and responsible AI practices.

Minimum Qualifications

• Master’s or Ph.D. in Computer Science, Data Science, AI, or a related field.

• Minimum of 3 (three) years of experience in AI/ML, with a proven track record of deploying solutions in a production environment.

• Proficiency in Python, R, or similar languages, with experience in libraries like TensorFlow, PyTorch, Scikit-learn, etc.

• Strong background in statistics, machine learning algorithms, deep learning, neural networks, and system architecture.

• Strong understanding of cloud platforms, DevOps practices, and data processing technologies.

• Demonstrated ability to lead technical projects and work effectively in matrix environments.

• Ability to communicate technical concepts to non-technical team members and business partners.

• Familiarity with software engineering best practices, including version control (Git), CI/CD pipelines, agile methodologies, and system design patterns.

About KLA

We provide advanced inspection tools, metrology systems, process solutions, and computational analytics that make electronics possible, tackling complex challenges. From electron and photon optics to machine learning and data analytics, we seek perfection at the most fundamental level of matter in the universe. If you want to make electronics that push industries forward and make the world a better place, join us.

Use of AI Statement 

At KLA, our interviews seek to understand your individual skills, problem-solving approach and authentic thinking. To ensure a fair and consistent evaluation, the use of AI, recording tools or other technologies to generate, suggest or provide responses during interviews—whether virtual or in person—is not permitted unless explicitly approved in advance as part of a reasonable accommodation or invited by the interviewer. Use of these tools may interfere with our ability to evaluate your individual qualifications and affect your candidacy. KLA is committed to advancing innovation through responsible AI, and we value candidates who share this mindset.

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.