Sr. Software Engineer - AI/ML

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

Benefits offered by this job

Medical benefits

401(k) with company matching

Tuition reimbursement

Employee assistance program

Job summary

KLA-Belgium is on the lookout for a Senior Software Engineer to enhance AI solutions within their technology team in Ann Arbor, MI. This critical role involves designing scalable machine learning models, deploying impactful AI applications, and collaborating with cross-functional teams.

The ideal applicant will have extensive experience in software engineering, particularly in AI/ML environments, and excellent communication skills to convey technical concepts to diverse audiences. A hybrid work arrangement provides flexibility for this position.

Qualifications

• 8+ years of software engineering experience, with 3–5 years in AI/ML.
• Proficiency in writing scalable applications in Python.
• Strong understanding of machine learning algorithms and statistics.

Responsibilities

• Design and implement machine learning models and AI algorithms.
• Architect scalable systems to handle AI/ML workloads.
• Collaborate with data engineers and product managers to integrate solutions.

Skills

Machine Learning

Artificial Intelligence

Data Science

Python

TensorFlow

Scikit-learn

Education

Bachelor's in Computer Science or related field

Master's or PhD preferred

Tools

Cloud Platforms

CI/CD Pipelines

DevOps

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

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

Minimum Qualifications

• 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 TensorFlow, 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, 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.
• Hybrid work arrangement; based in Ann Arbor, MI.

Benefits & Compensation

• Base Pay Range: $129,600.00 – $220,300.00 annually.
• Participation in performance incentive programs and eligibility for additional benefits: medical, dental, vision, life, and other voluntary benefits; 401(k) with company matching; employee stock purchase program (ESPP); student debt assistance; tuition reimbursement; development and career growth opportunities; financial planning benefits; wellness benefits including employee assistance program; paid time off and company holidays; family care and bonding leave.
• Hybrid role based at our Midwest HQ in Ann Arbor, MI.

Equal Opportunity Employer

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. For additional information, view the US Know Your Rights poster on the U.S. Equal Employment Opportunity Commission website. We do not accept resumes from headhunters, placement agencies, or other suppliers that have not signed a formal agreement with us.