Software Engineer - AI/ML

Software Engineer - AI/ML KLA paid time off, tuition reimbursement, 401(k) United States, Michigan, Ann Arbor 1200 Woodridge Avenue (Show on map) May 13, 2026 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 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 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.

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:

$105,900.00 - $180,000.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 attalent.acquisition@kla.com or at +1-408-352-2808 to request accommodation. Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees. KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA's Careers website for legitimate job postings. KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers. If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to talent.acquisition@kla.com to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.