Sr. Ai Infrastructure Software Engineer

KLA

Milpitas, CA, USA
$166,700.00 - $283,400.00 annually; not specified;...
C++ programming skills
Gpu programming (cuda)
Linux development environments
Design and implement core infrastructure components that support AI/ML workloads across multiple frameworks and hardware platforms

Job Summary

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

Matching Summary

Design and implement core infrastructure components that support AI/ML workloads across multiple frameworks and hardware platforms.

Salary

$166,700.00 - $283,400.00 Annually; Not specified; Not specified

Skills & Requirements

Must-have

  • C++ programming skills
  • GPU programming (CUDA)
  • Linux development environments
  • Build systems (Makefiles, CMake)
  • Compiler knowledge (Clang, GCC)

Nice-to-have

  • AI/ML frameworks (TensorFlow, PyTorch, JAX)
  • Embedded systems experience
  • Physical product development
  • Technical project coordination

Key Requirements

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

Work Rights

Not specified

Tailored Resume

Cover Letter