KLA is looking for a Software Engineer to build the backbone of AI innovation by developing and optimizing core infrastructure components for next-generation AI frameworks like CUDA, TensorFlow, PyTorch, and JAX, enabling scalable platforms for AI/ML integration within their semiconductor manufacturing solutions.
Requirements
- Strong proficiency in Morden 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.
- 3–5 years of experience in system-level software development.
- Experience with AI frameworks like 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.)
- Excellent problem-solving skills and a collaborative mindset.
- Work alongside visionary architects and engineers on AI infrastructure that spans multiple divisions.
- Grow your expertise in high-performance computing and contribute to real-world AI applications.
- Be part of a team that values innovation, mentorship, and continuous learning.
