Company Overview

KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Virtually every electronic device in the world is produced using our technologies. No laptop, smartphone, wearable device, voice-controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us. KLA invents systems and solutions for the manufacturing of wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. The innovative ideas and devices that are advancing humanity all begin with inspiration, research and development. KLA focuses more than average on innovation and we invest 15% of sales back into R&D. Our expert teams of physicists, engineers, data scientists and problem-solvers work together with the world’s leading technology providers to accelerate the delivery of tomorrow’s electronic devices. Life here is exciting and our teams thrive on tackling really hard problems. There is never a dull moment with us.

Group/Division

Enabling the movement toward advanced chip design, KLA's Measurement, Analytics and Control group (MACH) is looking for the best and brightest research scientists, software engineers, application development engineers and senior product technology process engineers to join our team. The MACH team's mission is to collaborate with our customers to innovate technologies and solutions that detect and control highly complex process variations—at their source—rather than compensate for them at later stages of the manufacturing process. With over 40 years of semiconductor process control experience, chipmakers around the globe rely on KLA to ensure that their fabs ramp next-generation devices to volume production quickly and cost-effectively. Our MACH team develops leading-edge solutions for patterning process analytics and control technologies, thereby providing customers with critical insight at the feature level, field level and cross-wafer analysis. Our teams also develop advanced modeling simulation, data analytics and process control modeling technologies. As a member of the MACH team, you’ll be joining the most sophisticated and successful process-control company in the semiconductor industry--working across functions to solve the most complex technical problems in the digital age.

Job Description/Preferred Qualifications

• Implement and optimize CUDA kernels for image operations: convolution/filters, morphological ops, warping/resampling, color space conversions (RGB/YUV/HSV), denoising/deblurring, HDR/Tone mapping, polygon manipulation, feature extraction, and classification.
• Use GPU memory hierarchies effectively (global/shared/constant/texture), coalesce memory, apply shared memory tiling, and minimize divergence/branching.
• Profile and tune with Nsight Compute/Systems, CUDA-MEMCHECK, and cuda-gdb; instrument pipelines with metrics (FPS, latency, bandwidth, occupancy).
• Collaborate with product and algorithm teams; contribute to CI/CD (Azure/DevOps, CMake, GitHub Actions/GitLab CI) and documentation.
• Integrate accelerated primitives (NVIDIA NPP, cuFFT, cuBLAS) and OpenCV CUDA modules; build clean C++ APIs with Python bindings (pybind11) when needed.- implement a distributed multi-process architecture using CUDA MPS for high-throughput, concurrent workloads. You’ll own performance-critical pipelines, profile on NVIDIA GPUs, and ship production-quality C++ that meets strict latency and throughput targets

Minimum Qualifications

Master's Level Degree and 2+ years related work experience; Bachelor's Level Degree and related work experience of 4+ years

Minimum qualifications- BS/MS in CS, EE, or related field.

• 3–6 years of professional experience, including 2+ years focused on CUDA-based image processing.
• Strong C++17/20 fundamentals; solid understanding of parallel algorithms and data layouts (pitch-linear, planar, interleaved).
• Practical experience with Nsight profiling, occupancy analysis, and kernel optimization (tiling, warp-level intrinsics, streams).
• Experience with OpenCV (including CUDA paths), and at least one of: NPP, cuFFT, cuBLAS.
• Comfortable on Linux; CMake, Git, code reviews, and automated testing.

Preferred qualifications- DMA/zero-copy, pinned memory.

• Texture memory and surface writes for image sampling; CUDA Graphs and stream concurrency.
• Familiarity with image feature extraction and classification- Exposure to ML inference acceleration (TensorRT/cuDNN) for CV tasks is a plus, but core focus is classical image processing.
• Experience with performance modeling (roofline), and multi-GPU awareness (NCCL) is a bonus.
• Experience with sharding, Tensor parallelism, etc

Base Pay Range: $114,800.00 - $195,200.00 Annually

Primary Location: USA-CA-Milpitas-KLA

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) 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, and family care and bonding leave.

Interns are eligible for some of the benefits listed. Our 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. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

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. Please contact us at talent.acquisition@kla.com or at +1-408-352-2808 to request accommodation.

Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees. KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA’s Careers website for legitimate job postings. KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers. If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to talent.acquisition@kla.com to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.