KLA is looking to solve the problem of building the backbone of AI innovation by developing next-generation software infrastructure that powers groundbreaking AI frameworks
Requirements
- 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.
- Strong grasp of modern AI trends, including model optimization, distributed training, and inference acceleration.
- Experience with AI accelerators or heterogeneous computing environments.
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.
- Collaborate with data scientists and AI infrastructure engineers to build and scale machine learning based solutions for semiconductor customers.
Other
- Advanced degree in Computer Science, Electrical Engineering, or a related field, with a strong background in system-level software development.
- Excellent written and verbal communication skills.
- Proven track record to develop scalable software infrastructure across multiple divisions or organizations.
- Strong collaboration skills, with the ability to influence and align partners across teams and geographies.
- 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
