Sr. AI Platform Architect

Reposted 21 Days Ago
Be an Early Applicant
Ann Arbor, MI
In-Office
108K-184K Annually
Senior level
Hardware • Semiconductor
The Role
The GenAI Platform Architect will design and manage AI/ML infrastructure, focusing on hybrid systems, and collaborate on LLMOps capabilities and performance monitoring.
Summary Generated by Built In

Company Overview

KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Virtually every electronic device in the world is produced using our technologies. No laptop, smartphone, wearable device, voice-controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us. KLA invents systems and solutions for the manufacturing of wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. The innovative ideas and devices that are advancing humanity all begin with inspiration, research and development. KLA focuses more than average on innovation and we invest 15% of sales back into R&D. Our expert teams of physicists, engineers, data scientists and problem-solvers work together with the world’s leading technology providers to accelerate the delivery of tomorrow’s electronic devices. Life here is exciting and our teams thrive on tackling really hard problems. There is never a dull moment with us.

Group/Division

The Information Technology (IT) group at KLA is involved in every aspect of the global business. IT’s mission is to enable business growth and productivity by connecting people, process, and technology. It focuses not only on enhancing the technology that enables our business to thrive but also on how employees use and are empowered by technology. This integrated approach to customer service, creativity and technological excellence enables employee productivity, business analytics, and process excellence.

Job Description/Preferred Qualifications

We are seeking a highly skilled and motivated AI Platform Architect to join our Corporate IT Cloud & DevOps team. In this role, you will be responsible for architecting, building, and managing our Enterprise GenAI Hybrid Platform, enabling scalable and secure AI/ML workloads across on-premises and cloud environments. You will play a key role in operationalizing LLMOps capabilities, integrating open-source tools, and supporting distributed training and inferencing at scale

  • 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

Minimum Qualifications

  • 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

  • This role will be based at out Midwest HQ in Ann Arbor, MI

    Base Pay Range: $108,100.00 - $183,800.00 Annually

    Primary Location: USA-MI-Ann Arbor-KLA

    KLA’s total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to: medical, dental, vision, life, and 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 and paid company holidays, and family care and bonding leave.

    Interns are eligible for some of the benefits listed. Our pay ranges are determined by role, level, and location. The range displayed reflects the pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

                   

    KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [email protected] or at +1-408-352-2808 to request accommodation.

    Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees.  KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA’s Careers website for legitimate job postings.  KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers.  If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to [email protected] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.

    Top Skills

    Ai/Ml
    Ceph
    Flyte
    Gpu
    Infiniband
    Kubeflow
    Kubernetes
    Linux
    Minio
    Mlflow
    Am I A Good Fit?
    beta
    Get Personalized Job Insights.
    Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

    The Company
    HQ: Milipitas, CA
    10,001 Employees

    What We Do

    KLA develops industry-leading equipment and services that enable innovation throughout the electronics industry. We provide advanced process control and process-enabling solutions for manufacturing wafers and reticles. In close collaboration with leading customers across the globe, our expert teams of physicists, engineers, data scientists and problem-solvers design solutions that move the world forward.

    Similar Jobs

    General Motors Logo General Motors

    Embedded Software Engineer

    Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
    Hybrid
    2 Locations
    165000 Employees

    General Motors Logo General Motors

    Administrative Assistant

    Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
    Hybrid
    Warren, MI, USA
    165000 Employees

    General Motors Logo General Motors

    Site HR/LR Lead - Customer Care and Aftersales

    Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
    Hybrid
    2 Locations
    165000 Employees

    General Motors Logo General Motors

    Launch Electrical Software Coordinator

    Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
    Hybrid
    Lake Orion, MI, USA
    165000 Employees

    Similar Companies Hiring

    Red 6 Thumbnail
    Virtual Reality • Software • Hardware • Defense • Aerospace
    Orlando, Florida
    155 Employees
    Blissway Thumbnail
    Transportation • Software • Machine Learning • Internet of Things • Hardware • Fintech • Computer Vision
    Denver, Colorado
    20 Employees
    Turion Space Thumbnail
    Software • Manufacturing • Information Technology • Hardware • Defense • Artificial Intelligence • Aerospace
    Irvine, CA
    150 Employees

    Sign up now Access later

    Create Free Account

    Please log in or sign up to report this job.

    Create Free Account