Full-time

AI Infrastructure Software Architect

Posted by KLA • Milpitas, California, United States

📍 Milpitas, California 🕒 June 10, 2026

Apply for this Job Similar Jobs

About the Role

Description
 /Preferred QualificationsAre you passionate about building the backbone of AI innovation? Join our team as a Sr. AI Infrastructure Software Engineer and help craft the next-generation software infrastructure that powers groundbreaking AI frameworks like CUDA, TensorFlow, PyTorch, and JAX!
This role is ideal for engineers who excel in system-level development and are eager to expand their impact in high-performance computing—driving innovation where complexity meets scale!
Key Responsibilities:
Architect and implement robust software infrastructure to support AI/ML workloads across multiple frameworks and hardware platforms.
Design and optimize system-level components usingC,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 unifie...
                

Job Details

Location Milpitas, California
Job Type Full-time
Category Computer Occupations
Posted June 10, 2026
Deadline July 20, 2026

Ready to Apply?

Submit your application today and take the next step in your career journey with KLA.

Apply Now