Algorithm Software Architecture

KLA Corporation
Milpitas, United States of America
9 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Intermediate
Compensation
$ 271K

Job location

Milpitas, United States of America

Tech stack

C++
Code Review
Nvidia CUDA
Continuous Integration
Linux
Distributed Systems
Python
OpenCL
Software Architecture
Software Engineering
Scripting (Bash/Python/Go/Ruby)
Gpu Programming
Containerization
Kubernetes
Low Latency
Slurm
Data Pipelines
Docker

Job description

We are seeking a Senior Algorithm Software Architect to lead design and delivery of GPUaccelerated, highperformance computing software. You will set architectural direction, coach engineers, and partner with product and domain experts to deliver scalable, reliable systems for largescale compute and data workflows., * Own the endtoend software architecture for HPC/GPU platforms (services, libraries, data pipelines, deployment)

  • Lead technical strategy, decision records, define candidate architectures, lead design reviews and drive decisions; drive clear tradeoffs for performance, reliability, and maintainability
  • Design and implement GPU kernels and frameworks (e.g., CUDA, OpenCL, NCCL), optimizing for throughput, latency, and memory use
  • Guide parallel and distributed computing patterns (MPI, multiGPU scaling, heterogeneous compute)
  • Establish performance engineering practices: profiling, benchmarking, regression performance gates (Nsight Systems/Compute, nvprof)
  • Collaborate across functions; convert requirements into clear technical plans, roadmaps, and measurable outcomes
  • Uphold engineering excellence: coding standards, code reviews, test strategies, observability, security considerations
  • Mentor engineers; provide technical leadership on design, delivery, and career growth.
  • Communicate architecture, risks, and status to executives and stakeholders with clarity and candor.

Requirements

  • 10+ years in software engineering; 5+ years in software architecture for HPC or largescale systems
  • Expert in C++ (17/20) and one scripting language (Python preferred)
  • GPU programming expertise (CUDA, OpenCL); strong knowledge of GPU memory hierarchies, streams, occupancy
  • Handson with parallel/distributed stacks (MPI, NCCL, gRPC) and Linux performance tooling
  • Experience with cluster orchestration (Slurm, Kubernetes), CI/CD, and containerization (Docker)
  • Track record of technical leadership and exceptional communication with crossfunctional teams.

Preferred / NicetoHave

  • Multinode, multiGPU scaling; mixed precision; numerical methods and algorithms.
  • Experience with H200/H100/A100/L40Sclass accelerators and modern profiling workflows., 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

Benefits & conditions

Base Pay Range: $159,500.00 - $271,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.

About the company

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 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. Enabling the movement towards advanced chip design, KLA's Global Products Group (GPG), which is responsible for creating all of KLA's metrology and inspection products, is looking for the best and the brightest research scientist, software engineers, application development engineers, and senior product technology process engineers. The Film and Scatterometry Technology (FaST) Division provides industry leading metrology solutions for worldwide semiconductor IC manufacturers. The FaST Division portfolio of metrology products includes hardware and software solutions for optical film thickness, optical critical dimension (CD), composition, and resistivity measurement systems. These products are essential for the IC manufacturers as they provide critical metrology capabilities for the development and implementation of their advanced IC processes. The FaST division is committed to support our customers to achieve performance entitlement of our solution and we effectively partner with our customers from their early research and development phase to the high volume in-line manufacturing implementation specific for their process needs. The division consists of a global team located in US, Israel, China, and India.

Apply for this position