Sr. Software Engineer - AI/ML

KLA; Ann Arbor (MI); Hybrid; USD 150,000 - 210,000; Full time

Job summary

KLA, a global leader in semiconductor manufacturing, seeks a Sr. Software Engineer specialized in AI/ML to join its IT team at the Midwest HQ in Ann Arbor, MI. You will design and deploy AI solutions, collaborate with data engineers, PMs, and software teams, and ensure scalable, production-ready results.

The role emphasizes staying current with AI advances, implementing robust architectures, and contributing to data processing, security, and DevOps practices in a hybrid work environment.

Qualifications

• Bachelor's degree in Computer Science, Data Science, AI, or related field.

Responsibilities

• Design, develop, and implement AI/ML models and algorithms.

Skills

Technical leadership

AI/ML expertise

Data science

Education

Bachelor's in CS/DS/AI

Masters or PhD preferred

Tools

Python

TensorFlow

Scikit-Learn

RAG

CI/CD

Git

Job description

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.

Group/Division

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.

Job Description/Preferred Qualifications

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

Minimum Qualifications

• Bachelor’s degree in Computer Science, Data Science, AI, or a related field. Master’s and/or PhD preferred

• Minimum of eight (8) years of software engineering, with 3-5 years of experience in AI/ML, with a proven track record of deploying solutions in a production environment.

• Proficiency in Python, or similar languages, with experience in libraries like TensorFlow, Scikit-learn, etc. Experience with RAG is essential.

• 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.

• This is a hybrid role and will be based out of our Midwest HQ in Ann Arbor, MI.

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.

Base Pay Range: $129,600.00 - $220,300.00 Annually

Primary Location: USA-MI-Ann Arbor-KLA

KLA’s total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to: medical, dental, vision, life, and other voluntary benefits, 401(K) including company matching, employee stock purchase program (ESPP), student debt assistance, tuition reimbursement program, development and career growth opportunities and programs, financial planning benefits, wellness benefits including an employee assistance program (EAP), paid time off and paid company holidays, and family care and bonding leave.

Interns are eligible for some of the benefits listed. Our pay ranges are determined by role, level, and location. The range displayed reflects the pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [e-mail] or at [phone] to request accommodation.