Sr. AI Software Engineer (GPU/C++)

KLA-Belgium; Milpitas (CA); On-site; USD 166,700 - 283,400; Full time

Benefits offered by this job

Medical, dental, vision benefits

401(K) with company matching

Tuition reimbursement program

Wellness benefits

Job summary

KLA-Belgium in Milpitas seeks a Senior AI Infrastructure Software Engineer with a focus on C++. You will design core infrastructure components for AI/ML workloads and write high-performance C++ code interfacing directly with hardware.

The position requires strong C++ skills, GPU programming expertise, and experience in collaborative environments. KLA offers a comprehensive benefits package and opportunities for career growth.

Qualifications

• Minimum qualifications: Doctorate and 3 years of related work experience, or Master's level degree and 6 years, or Bachelor's degree and 8 years.
• Strong C++ programming skills and solid software engineering fundamentals.
• Familiarity with build tools and compilers.

Responsibilities

• Design and implement core infrastructure components for AI/ML workloads.
• Write high-performance, maintainable C++ code for system-level software.
• Collaborate with cross-functional teams to integrate AI frameworks.

Skills

C++ programming skills

GPU programming (e.g., CUDA)

Linux development environments

Communication skills

Attention to detail

Education

Doctorate in Computer Science, Electrical Engineering or related field

Master's degree and 6 years of experience

Bachelor's degree and 8 years of experience

Tools

Makefile

CMake

Clang

GCC

AI/ML frameworks (e.g. TensorFlow, PyTorch)

Job description

Job Title

Senior AI Infrastructure Software Engineer – C++ Focus

Responsibilities

• Design and implement core infrastructure components that support AI/ML workloads across multiple frameworks and hardware platforms.
• Write high-performance, maintainable C++ code for system‑level software that interfaces directly with GPUs and low‑level hardware.
• Collaborate with cross‑functional teams to integrate AI frameworks into a unified development environment.
• Contribute to build systems and toolchains using Makefiles, CMake, Clang, and GCC.
• Learn from experienced engineers and grow into broader architectural and leadership responsibilities over time.

Qualifications

• Degree in Computer Science, Electrical Engineering, or a related field.
• Strong C++ programming skills and solid software engineering fundamentals.
• Solid experience and hands‑on knowledge with GPU programming (e.g., CUDA) and Linux development environments.
• Familiarity with build tools (Makefile, CMake) and compilers (Clang, GCC).
• Bonus: Exposure to AI/ML frameworks such as TensorFlow, PyTorch, or JAX.
• Bonus: Experience with embedded systems or companies building physical products.
• Excellent attention to detail, planning, and execution skills—especially for multi‑month projects.
• Strong communication skills and a collaborative, customer‑focused mindset.
• A plus if you’ve transitioned from software development into technical project coordination or aspire to do so.
• Minimum qualifications: Doctorate (Academic) and 3 years of related work experience, or Master’s level degree and 6 years of experience, or Bachelor’s degree and 8 years of experience.

Eligibility & Compensation

Base pay range: $166,700.00 – $283,400.00 annually. Primary location: USA-CA-Milpitas, KLA. Compensation is determined by role, level, and location and is subject to state minimum wage requirements.

Benefits

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) with company matching, employee stock purchase program (ESPP), student debt assistance, tuition reimbursement program, development and career growth opportunities, 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.

EEO Statement

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.