KLA is looking to solve the problem of building the backbone of AI innovation by developing and optimizing core infrastructure components for cross divisions using C++, Python, and Linux, and integrating AI/ML frameworks into scalable platforms.
Requirements
- Strong proficiency in Modern C++, Python, and Linux-based development.
- Familiarity with AI/ML frameworks (e.g., TensorFlow, PyTorch) and experience with GPU programming (CUDA is a plus).
- Solid knowledge on modern DL models.
- Solid experience on compiler toolchains, containers, and Kubernetes.
- Experience with CUDA, TensorFlow, PyTorch, and JAX.
Responsibilities
- Develop and optimize core infrastructure components that will be used for cross divisions using C++, Python and Linux.
- Collaborate with senior architects and cross-functional teams to integrate AI/ML frameworks into scalable platforms.
- Contribute to build systems and toolchains using Makefiles, CMake, Clang, and GCC.
- Prototype and validate performance-critical modules for AI workloads.
- Stay current with emerging trends in AI infrastructure and contribute ideas to improve system robustness and scalability.
Other
- Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (Ph.D preferred).
- 3–5 years of experience in system-level software development.
- Excellent problem-solving skills and a collaborative mindset.
- Minimum Qualifications: Master's Level Degree and related work experience of 3 years; Bachelor's Level Degree and related work experience of 5 years
- Travel requirements not specified
