KLA · 1 day ago
Sr. AI Platform Architect
Ann Arbor, MI
Full-time
Onsite
Senior Level, Lead/Staff
$108K/yr - $184K/yr
8+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. They are seeking a highly skilled AI Platform Architect to architect, build, and manage their Enterprise GenAI Hybrid Platform, enabling scalable and secure AI/ML workloads across on-premises and cloud environments.
ElectronicsInformation TechnologyManufacturing
Hiring Manager
Poonkodi Siva
Responsibilities
Design, deploy, and manage scalable AI/ML infrastructure supporting hybrid cloud and on-prem environments
Work extensively with open-source MLOps platforms (e.g., Kubeflow, MLflow, Flyte) to streamline model development, deployment, and lifecycle management
Architect and optimize GenAI infrastructure, including integration of vector databases and large language model serving frameworks
Implement and manage high-performance shared storage systems (e.g., Ceph, MinIO) for distributed AI workloads
Set up and maintain InfiniBand networking for low-latency, high-throughput GPU cluster communication
Collaborate with ML engineers, data scientists, and DevOps teams to build a cohesive and efficient AI/ML ecosystem
Monitor and enhance infrastructure performance, ensuring scalability, reliability, and security
Evaluate and integrate emerging GenAI tools and frameworks to continuously improve platform capabilities
Qualification
Required
Bachelor's Degree or equivalent experience in Computer Science or related IT field
Eight (8) years of implementing and maintaining AI/ML Infrastructure in an On-Prem environment
Strong experience with AI/ML infrastructure and tools, including GPU clusters and Kubernetes
Proficiency in deploying and managing open-source GenAI components and vector databases
Hands-on experience with high-performance computing (HPC) environments
Expertise in designing and managing on-premises, cloud, and hybrid-based ML platforms
Strong Linux system administration and scripting skills
Benefits
Medical
Dental
Vision
Life
Other voluntary benefits
401(K) including company matching
Employee stock purchase program (ESPP)
Student debt assistance
Tuition reimbursement program
Development and career growth opportunities and programs
Financial planning benefits
Wellness benefits including an employee assistance program (EAP)
Paid time off
Paid company holidays
Family care and bonding leave
Company
KLA
Kla creates tools and services that promote innovation in the electronics industry.
10001+ employees
H1B Sponsorship
KLA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)
Funding
Current Stage
Late StageLeadership Team
Lisa Grunwald
FISH Philosophy CEO
Rick Wallace
President & CEO
Recent News
news.com.au — Australia’s leading news site for latest headlines
2025-08-01
2025-08-01
Company data provided by crunchbase