Beyond Pre-Training: The Full Lifecycle of Foundation Models on HPC Systems

TL;DR

This paper explores the full AI lifecycle on HPC systems, proposing a hybrid cloud-native platform at Swiss National Supercomputing Centre to enable efficient fine-tuning and inference workflows.

Contribution

It introduces a novel Kubernetes-based architecture combining HPC and cloud resources for complete AI lifecycle management on supercomputers.

Findings

Hybrid platform improves user productivity in AI workflows

Analysis of trade-offs in fine-tuning pipelines and inference services

Blueprint for integrating AI services into supercomputing environments

Abstract

Large-scale pre-training of Foundational Models (FM) constitutes a computationally intensive first phase for enabling AI across diverse scientific and societal applications. This first phase has positioned High-Performance Computing (HPC) facilities as indispensable backbones of "Sovereign AI" initiatives. While the massive throughput requirements of FM pre-training align with the traditional capability-oriented mission of HPC, subsequent phases of the AI lifecycle, typically referred to as fine-tuning and inference, introduce operational paradigms that can conflict with established batch-processing environments. Moreover, these phases are not computationally trivial: they often require substantial high-end compute resources while exhibiting hardware utilization patterns that differ significantly from those of pre-training. This paper addresses the architectural and strategic challenges of operationalizing a complete AI lifecycle within a national supercomputing facility. We present a hybrid cloud-native platform being developed and deployed at the Swiss National Supercomputing Centre (CSCS) that combines diskless GPU-enabled HPE Cray EX compute nodes with virtualized commodity infrastructure. Orchestrated by Kubernetes, this novel service architecture bridges the gap between HPC batch processing and service-oriented workflows. We report our initial investigations into fine-tuning pipelines and highly available inference services, analyzing the associated trade-offs while improving user productivity. Our findings offer a blueprint for enabling supercomputers to integrate "AI Factories" services and workflows, supporting AI innovations into end-to-end scientific and industrial use cases.