✦ Luna Orbit — Software Engineering

Sr. AI Infrastructure Software Engineer

at KLA

📍 USA-CA-Milpitas-KLA Unknown 💰 $166K – $283K USD / year Posted March 13, 2026
Salary $166K – $283K USD / year
Type Not Specified
Experience mid
Exp. Years Not specified
Education Not specified
Category Software Engineering

This role involves developing advanced AI infrastructure software for semiconductor manufacturing, focusing on GPU acceleration and physics-based modeling.

  • Build next-gen AI software stack
  • Optimize GPU-based AI computations
  • Develop physics modeling solutions
  • Collaborate with physics and data scientists
  • Scale AI frameworks for hardware acceleration

The position requires expertise in C++, GPU programming, AI frameworks such as TensorFlow and PyTorch, and high-performance computing environments.

The ideal candidate is a curious and skilled software engineer with strong C++ expertise, experience with GPU programming, and familiarity with AI frameworks like TensorFlow and PyTorch. They should have a passion for physics modeling and high-performance computing in a semiconductor environment.

C++GPUCUDADeep LearningAI frameworksSoftware stack
TensorFlowJAXPyTorchPhysics ModelingHPC
CUDATensorFlowJAXPyTorchGPU
C++CUDATensorFlowJAXPyTorchGPUHPCPhysics ModelingDeep LearningAI frameworksSoftware stack
C++C++ programmingCUDATensorFlowJAXPyTorchGPUHigh Performance ComputingHPCPhysics ModelingDeep LearningAI frameworksSoftware stack
curiosityproblem-solvingteamworkgrowth mindsetcollaboration
Industry Semiconductor Manufacturing
Job Function Developing AI infrastructure software for semiconductor manufacturing
C++CUDATensorFlowJAXPyTorchGPUHigh Performance ComputingHPCPhysics ModelingDeep LearningAI frameworksSoftware stack

Lack of C++ experience, No GPU or CUDA experience, No familiarity with AI frameworks, Lack of experience in high-performance computing

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile