Demystifying AI Platform Design for Distributed Inference of Next-Generation LLM models

TL;DR

This paper introduces GenZ, an analytical tool that models and predicts hardware requirements for efficient deployment of diverse large language models, aiding in the design of next-generation AI platforms.

Contribution

The paper presents GenZ, a novel analytical framework that estimates LLM inference performance and hardware needs, validated against real hardware, to guide platform and hardware design.

Findings

GenZ achieves a maximum geomean error of 5.82 in performance estimation.

It identifies key hardware requirements like compute, memory, and network for various LLM use cases.

Insights from GenZ inform AI deployment strategies and hardware architecture design.

Abstract

Large language models (LLMs) have shown remarkable performance across a wide range of applications, often outperforming human experts. However, deploying these gigantic models efficiently for diverse inference use cases requires carefully designed hardware platforms with ample computing, memory, and network resources. With constant innovation in LLM serving optimizations and model architecture evolving at breakneck speed, the hardware requirements to meet Service Level Objectives (SLOs) remain an open research question. To answer the question, we present an analytical tool, GenZ, to efficiently navigate the relationship between diverse LLM model architectures(Dense, GQA, MoE, Mamba), LLM serving optimizations(Chunking, Speculative decoding, quanitization), and AI platform design parameters. Our tool estimates LLM inference performance metrics for the given scenario. We have validated against real hardware platforms running various different LLM models, achieving a max geomean error of 5.82.We use GenZ to identify compute, memory capacity, memory bandwidth, network latency, and network bandwidth requirements across diverse LLM inference use cases. We also study diverse architectural choices in use today (inspired by LLM serving platforms from several vendors) to help inform computer architects designing next-generation AI hardware accelerators and platforms. The trends and insights derived from GenZ can guide AI engineers deploying LLMs as well as computer architects designing next-generation hardware accelerators and platforms. Ultimately, this work sheds light on the platform design considerations for unlocking the full potential of large language models across a spectrum of applications. The source code is available at https://github.com/abhibambhaniya/GenZ-LLM-Analyzer . Users can also be tried it on at https://genz-llm-analyzer.streamlit.app/ without any setup on your web browser.