KLA

Senior AI Infrastructure Software Engineer

KLA

full-time

Posted on:

Location Type: Office

Location: Milpitas • California, Missouri • 🇺🇸 United States

Visit company website
✨ AI Apply
Apply

Salary

💰 $136,300 - $231,700 per year

Job Level

Senior

Tech Stack

KubernetesLinuxPythonPyTorchTensorflow

About the role

  • Develop and optimize core infrastructure components that will be used for cross divisions using C++, Python and Linux
  • Collaborate with senior architects and cross-functional teams to integrate AI/ML frameworks into scalable platforms
  • Contribute to build systems and toolchains using Makefiles, CMake, Clang, and GCC
  • Prototype and validate performance-critical modules for AI workloads
  • Stay current with emerging trends in AI infrastructure and contribute ideas to improve system robustness and scalability

Requirements

  • Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (Ph.D preferred)
  • 3–5 years of experience in system-level software development
  • Strong proficiency in Modern C++, C++, Python, and Linux-based development
  • Familiarity with AI/ML frameworks (e.g., TensorFlow, PyTorch) and experience with GPU programming (CUDA is a plus)
  • Solid knowledge on modern DL models
  • Solid experience on compiler toolchains, containers, and Kubernetes
  • Excellent problem-solving skills and a collaborative mindset
Benefits
  • medical
  • dental
  • vision
  • life
  • 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
  • paid company holidays
  • family care and bonding leave

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
C++PythonLinuxAI/ML frameworksTensorFlowPyTorchGPU programmingCUDAcompiler toolchainsKubernetes
Soft skills
problem-solvingcollaborative mindset
Certifications
Bachelor’s degreeMaster’s degreePh.D