AI Infrastructure Engineer: Build Scalable ML Platforms

KLA-Belgium; Milpitas (CA); On-site; USD 136,300 - 231,700; Full time

Benefits offered by this job

Medical benefits

401(k) matching

Employee stock purchase program

Tuition reimbursement

Job summary

KLA is seeking an AI Infrastructure Engineer to design and operate scalable AI platforms supporting large-scale model training and inference. You will work across AI research, data science, and IT to deliver reliable, high-performance infrastructure for ML workloads.

You’ll evaluate new technologies like GPUs, distributed training, and orchestration tools, while ensuring security-by-design and operational excellence across KLA products.

Qualifications

• Degree in Computer Science, Computer Engineering, or related field.
• 5–8 years in systems engineering, DevOps, or ML infrastructure.
• Hands-on experience building AI/GPU clusters.

Responsibilities

• Design, build, and operate scalable AI platforms and model serving environments.
• Research and evaluate emerging AI infrastructure technologies.
• Ensure security, observability, and reliability of AI workloads.

Skills

AI infrastructure

DevOps experience

GPU cluster

Cloud computing

Security by design

Education

Bachelor's degree in Computer Science/Engineering

Master's degree (preferred)

PhD in related field (optional)

Tools

Kubernetes

Docker

Slurm

Ray

TensorFlow

PyTorch

Job description

KLA is seeking an AI Infrastructure Engineer to design and operate scalable AI platforms supporting large-scale model training and inference. You will work across AI research, data science, and IT to deliver reliable, high-performance infrastructure for ML workloads.

You’ll evaluate new technologies like GPUs, distributed training, and orchestration tools, while ensuring security-by-design and operational excellence across KLA products.