Software Engineer - AI/ML

KLA-Belgium; Town of Belgium (WI); On-site; USD 79,926 - 125,599; Full time

Job summary

KLA in Leuven, Belgium seeks a Software Engineer specialized in AI/ML to design, develop, and deploy machine learning models and AI algorithms that address real business challenges. You will build scalable architectures, collaborate with data engineers, product managers, and software developers, and ensure solutions are robust, secure, and production-ready.

The role requires a Master’s or PhD and hands-on experience with TensorFlow/PyTorch, Python, and cloud platforms, with a focus on

Qualifications

• 3+ years of AI/ML experience in production environments.
• Proficiency in Python (or similar) with ML libraries.
• Strong background in ML algorithms, deep learning, and system design.

Responsibilities

• Design, develop, and implement ML models and AI algorithms.
• Architect scalable AI/ML systems for large data volumes.
• Deploy production-ready solutions with security and reliability in mind.
• Collaborate with data engineers, product managers, and software teams.
• Monitor, retrain, and improve models post-deployment.

Skills

Python

TensorFlow

PyTorch

Scikit-learn

Keras

Statistics

Git

CI/CD

Communication

Education

Master’s or Ph.D. in Computer Science / Data Science / AI or related field

Tools

TensorFlow

PyTorch

Scikit-learn

Keras

Cloud (AWS/Azure/GCP)

Job description

## Software Engineer - AI/MLApplylocations: Leuven, Belgiumtime type: Full timeposted on: Posted Yesterdayjob requisition id: 2638258**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.**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.*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*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 [e-mail] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.