→ Back to Home
Generative AI

Achieving Multi-AZ Resiliency for Shared GPU LLM Inference with SageMaker

AWS and Salesforce detailed how Salesforce architected Multi-AZ high availability for Agentforce foundation models using Amazon SageMaker AI Inference Components (ICs). By leveraging the SchedulingConfig parameter within the CreateInferenceComponent API, Salesforce resolved an availability challenge: co-hosting multiple models on shared GPUs delivered an 8x infrastructure cost reduction, but SageMaker’s default placement algorithm placed component copies across instances without zone awareness, leaving critical models exposed to single-instance or single-AZ outages. Using fine-grained controls including AvailabilityZoneBalance and the SPREAD placement strategy, Salesforce achieved strict multi-zone compliance without forfeiting multi-tenant efficiency. For MLOps architects and DevOps engineers, running generative AI workloads at scale introduces conflicting architectural pressures. Standard single-model dedicated hosting ensures clear fault domains and multi-AZ load balancing but generates massive GPU underutilization and unsustainable inference bills. While SageMaker Inference Components enabled multi-model packing onto shared accelerators, early implementations lacked granular topology awareness. An entire model’s replicas could accidentally concentrate within a single failure domain. The addition of explicit scheduling policies makes multi-tenant model serving viable for tier-one, mission-critical generative AI systems subject to stringent recovery-point and uptime service level agreements. This evolution mirrors the trajectory of Kubernetes pod topology spread constraints and affinity rules in traditional container orchestration. As AI workloads transition from experimental chatbot endpoints to continuous background agents, infrastructure tooling must evolve from rudimentary instance-level provisioning to cluster-aware, zone-aware resource scheduling. The move by Salesforce and AWS highlights that generative AI infrastructure is maturing from simple compute consumption into sophisticated capacity engineering where bin-packing efficiency, zonal failure isolation, and dynamic scaling must coexist. In practice, engineering teams implementing SageMaker Inference Components for high availability must review their deployment manifests and avoid default scheduling assumptions. Workloads requiring multi-AZ resilience should explicitly set AvailabilityZoneBalance with low MaxImbalance tolerances and adopt SPREAD over BINPACK to isolate copies across distinct hardware nodes. Furthermore, teams must ensure baseline replica counts are at least two to satisfy multi-zone placement, and leverage On-Demand Capacity Reservations (ODCR) to prevent zonal capacity shortages from blocking rebalancing during instance recovery or rolling model deployments.
#sagemaker#generative ai#model serving#high availability#mlops
Read original source