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

About the role

The AI & Modeling Center of Excellence within KLA is seeking a Senior AI Infrastructure Software Engineer with a focus on C++. This role involves designing and implementing core infrastructure components that support AI/ML workloads across multiple frameworks and hardware platforms. You will write high-performance, maintainable C++ code for system-level software that interfaces directly with GPUs and low-level hardware. Collaboration with cross-functional teams is key, as is contributing to build systems and toolchains using Makefiles, CMake, Clang, and GCC.

Responsibilities

• Design and implement core infrastructure components for 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.

Requirements

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

Qualifications

• Doctorate (Academic) Degree and related work experience of 3 years;
• Master's Level Degree and related work experience of 6 years;
• Bachelor's Level Degree and related work experience of 8 years;

Skills

• Experience with AI/ML frameworks like TensorFlow, PyTorch, or JAX.
• Experience with embedded systems or companies building physical products.

Benefits

• Base Pay Range: $166,700.00 - $283,400.00 Annually
• Additional benefits include:
• Medical, dental, vision, life, and other voluntary benefits
• 401(K) including company matching
• Employee stock purchase program (ESPP)
• Student debt assistance
• Tuition reimbursement program
• Development and career growth opportunities and programs
• Financial planning benefits
• Wellness benefits including an employee assistance program (EAP)
• Paid time off and paid company holidays
• Family care and bonding leave

Pay

• Pay ranges are determined by role, level, and location.
• The range displayed reflects the pay for this position in the primary location identified in this posting.
• Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training.

Contact Information

• To request accommodation, please email talent.acquisition@kla.com or call +1-408-352-2808.
• To report suspicious recruiting activity, please email talent.acquisition@kla.com.