KLA is looking for a Software Engineer to build the backbone of AI innovation by developing and optimizing core infrastructure components for next-generation software infrastructure that powers cutting-edge AI frameworks.
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.
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.
