KLA · 11 hours ago
AI Infrastructure Software Architect
Milpitas, CA
Full-time
Onsite
Senior Level
$180K/yr - $306K/yr
5+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem, seeking a Senior AI Infrastructure Software Engineer. The role involves architecting and implementing robust software infrastructure to support AI/ML workloads across multiple frameworks and hardware platforms while providing technical leadership and mentorship to junior engineers.
ElectronicsInformation TechnologyManufacturing
Hiring Manager
Keziah Catherene Abraham
Responsibilities
Architect and implement robust software infrastructure to support AI/ML workloads across multiple frameworks and hardware platforms
Design and optimize system-level components using C, C++, and Python, ensuring high performance, scalability, and maintainability
Collaborate with multi-functional and cross-divisional teams to integrate AI frameworks (e.g., TensorFlow, PyTorch, JAX, ONNX) into a unified development environment
Lead the development of build systems and toolchains using Makefiles, CMake, Clang, and GCC
Stay ahead of modern software and AI trends, and guide the team in adopting standard processes and new technologies
Provide technical leadership and mentorship to junior engineers and chip in to long-term architectural decisions
Qualification
Required
Advanced degree in Computer Science, Electrical Engineering, or a related field, with a strong background in system-level software development
Proficient in C, modern C++, Python, and experienced with Linux development environments
Familiar with build tools (Makefile, CMake), compilers (Clang, GCC), and container technologies (e.g., Kubernetes)
Deep understanding of software architecture, compiler toolchains, and runtime systems
Experience building infrastructure for AI/ML frameworks or high-performance computing environments
Hands-on experience with CUDA, GPU programming, and familiarity with TensorFlow and deep learning model interoperability is a strong plus
Proven track record to develop scalable software infrastructure across multiple divisions or organizations
Strong grasp of modern AI trends, including model optimization, distributed training, and inference acceleration
Excellent written and verbal communication skills
Doctorate (Academic) Degree and related work experience of 5 years; Master's Level Degree and related work experience of 8 years; Bachelor's Level Degree and related work experience of 12 years
Preferred
Experience working with AI accelerators or heterogeneous computing environments
Contributions to open-source AI or systems software projects
Strong collaboration skills, with the ability to influence and align partners across teams and geographies
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
Company
KLA
Kla creates tools and services that promote innovation in the electronics industry.
10001+ employees
H1B Sponsorship
KLA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)
Funding
Current Stage
Late StageLeadership Team
Lisa Grunwald
FISH Philosophy CEO
Rick Wallace
President & CEO
Recent News
news.com.au — Australia’s leading news site for latest headlines
2025-08-01
2025-08-01
Company data provided by crunchbase