Sr. Software Engineer - AI/ML

KLA; Ann Arbor (MI); Hybrid; USD 129,600 - 220,300; Full time

Benefits offered by this job

401(K) with company matching

Employee stock purchase program (ESPP)

Tuition reimbursement

Paid time off

Family care leave

Job summary

KLA is seeking a highly skilled Sr. Software Engineer focused on Machine Learning, Artificial Intelligence, and Data Science. This role entails developing impactful AI solutions that drive business transformation.

The successful candidate will design and implement machine learning models and AI algorithms, oversee scalable system architectures, and collaborate with teams to integrate solutions. A Bachelor’s degree and extensive experience in software engineering and AI/ML are essential. This position is hybrid based in Ann Arbor, MI.

Qualifications

• 8+ years of software engineering experience, with 3–5 years in AI/ML.
• Proficiency in Python or similar languages.
• Strong background in machine learning algorithms and neural networks.

Responsibilities

• Design and implement machine learning models and AI algorithms.
• Architect scalable systems for AI/ML workloads.
• Document processes and share knowledge through workshops.

Skills

Machine Learning

Artificial Intelligence

Data Science

Python

TensorFlow

Deep Learning

Education

Bachelor’s degree in Computer Science, Data Science, AI, or related field

Master’s and/or PhD preferred

Tools

Scikit-learn

Cloud platforms

DevOps practices

Job description

Job Summary

We are seeking a highly skilled Sr. Software Engineer with specialized expertise in Machine Learning, Artificial Intelligence, and Data Science to join our technology team. The role focuses on developing impactful AI solutions that drive business transformation and technological advancement.

Key Responsibilities

• Design, develop, and implement machine learning models and AI algorithms to address business challenges, including data preprocessing, model training, validation, and deployment.
• Architect scalable systems capable of handling large volumes of data and complex AI/ML workloads, ensuring high performance and reliability.
• Implement robust, maintainable, production‑ready AI/ML solutions with considerations for security, compliance, and operational efficiency.
• Stay abreast of the latest developments in AI, ML, and data science; conduct research to explore new methodologies or improve existing ones.
• Collaborate closely with cross‑functional teams (data engineers, product managers, software developers, business partners) to integrate advanced solutions into products and services.
• Apply critical thinking to solve complex problems requiring innovative approaches to data analysis, model architecture, system integration, and business strategy.
• Oversee deployment of AI/ML models into production environments, ensuring scalability, performance, reliability, and post‑deployment monitoring.
• Document processes, methodologies, and findings; share knowledge through internal presentations, workshops, or contributions to the knowledge base.

Minimum Qualifications

• Bachelor’s degree in Computer Science, Data Science, AI, or related field; Master’s and/or PhD preferred.
• Minimum eight (8) years of software engineering experience, with 3–5 years in AI/ML and a proven record of deploying solutions in production.
• Proficiency in Python or similar languages; experience with TensorFlow, Scikit‑learn, and RAG essential.
• Strong background in statistics, machine learning algorithms, deep learning, neural networks, and system architecture.
• Solid understanding of cloud platforms, DevOps practices, and data processing technologies.
• Demonstrated ability to lead technical projects and work effectively in matrix environments.
• Excellent communication skills for translating technical concepts to non‑technical stakeholders.
• Familiarity with software engineering best practices, version control (Git), CI/CD pipelines, agile methodologies, and design patterns.
• Hybrid role based out of our Midwest HQ in Ann Arbor, MI.

Preferred Qualifications

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

Compensation and Benefits

Base Pay Range: $129,600.00 – $220,300.00 Annually (salary depends on role, level, and location). Additional benefits may include performance incentive programs, 401(K) with company matching, employee stock purchase program (ESPP), student debt assistance, tuition reimbursement, development and career growth programs, financial planning benefits, wellness benefits, paid time off, paid holidays, and family care leave.

Location

USA – MI – Ann Arbor – KLA

Equal Opportunity Employer

KLA is proud to be an Equal Opportunity Employer. We will ensure qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process. Please contact [e-mail] for accommodation requests.