Deep dive on LLM Inference at Scale โ Harshul Jain, Audible & Tanmay Sah, Independent AI Researcher
ํ ์ค ํต์ฌ
LLM ์ถ๋ก ๋น์ฉ์ ๋ฉ๋ชจ๋ฆฌ, ์ฒซ ํ ํฐ ์ง์ฐ, ์ฒ๋ฆฌ๋์ ์ ์ฝ์์ ๋น๋กฏ๋๋ฉฐ, ๋ชจ๋ธ ์์ถ๊ณผ ์๋น ์ต์ ํ๋ฅผ ํจ๊ป ์ ์ฉํด GPU๋น ์์ฉ๋๊ณผ ๋น์ฉ ํจ์จ์ ๋์์ ๋์ด์ฌ๋ ค์ผ ํ๋ค.
์๊ฐ๋ณ ์์ฝ
-
0:00
์๊ฐ์ ์ํฌ์ ๋ฒ์ LLM inference๋ฅผ ์ฒ์๋ถํฐ ์ดํดํ๊ณ ์ ๊ณ์์ ์ค์ ๋ก ๋ฌด์์ด ๋ณ๋ชฉ์ธ์ง ํ์ ํ๋ ๊ฒ์ด ๋ชฉํ์ด๋ฉฐ, ์ด๊ธ๋ถํฐ ์ค๊ธ๊น์ง๋ฅผ ๋์์ผ๋ก ๋ฆฌํฌ์งํ ๋ฆฌ์ ์ฌ๋ผ์ด๋์ ์ค์ต ์๋ฃ๋ฅผ ํจ๊ป ํ์ฉํ๋ค.
-
2:43
LLM ์ถ๋ก ์ด ๋ฌด์์ด๋ฉฐ ์ ๋น์ฉ์ด ํฐ๊ฐ LLM inference๋ ํ ์คํธ, ์ค๋์ค, ์์ ๋ถ์ ๊ฐ์ ์์ฒญ์ ์ฒ๋ฆฌํ๋ ์คํ ๋จ๊ณ์ด๋ฉฐ, ํ์ต๋ณด๋ค ํจ์ฌ ํฐ ๋ฐ๋ณต ์ด์๋น๊ฐ ๋ค์ด์ ํ ํฐ ์ฌ์ฉ๋๊ณผ ์ถ๋ก ํจ์จ์ ๊ด๋ฆฌํ์ง ์์ผ๋ฉด ๋น์ฆ๋์ค๊ฐ ์์ต๋ถ๊ธฐ์ ์ ๋๊ธฐ๊ธฐ ์ด๋ ต๋ค.
-
5:21
๋ฆฌํฌ์งํ ๋ฆฌ, ์ฌ๋ผ์ด๋, ๋ฌด๋ฃ GPU ๋ ธํธ๋ถ ๊นํ๋ธ ๋ฆฌํฌ์งํ ๋ฆฌ์๋ ์ฌ๋ผ์ด๋, ๋ฒค์น๋งํฌ ๋ณด๊ณ ์, Jupyter notebook์ด ์ ๋ฆฌ๋์ด ์๊ณ , ๋ฌด๋ฃ RTX 6000 GPU ํ๊ฒฝ์ ํตํด ์ค์ ๋ก ๋ชจ๋ธ์ ์ฌ๋ฆฌ๊ณ GPU ๋ฉ๋ชจ๋ฆฌ์ ์คํ ๊ฒฐ๊ณผ๋ฅผ ํ์ธํ ์ ์๋ค.
-
8:23
์ธ ๊ฐ์ง ๋ณ๋ชฉ: ๋ฉ๋ชจ๋ฆฌ, ์ฒซ ํ ํฐ, ์ฒ๋ฆฌ๋ ์ปจํ ์คํธ๊ฐ ๊ธธ์ด์ง์๋ก KV cache ๋๋ฌธ์ ๋ฉ๋ชจ๋ฆฌ๊ฐ ๋๊ณ , prefill์ด ๊ธธ์ด์ ธ TTFT๊ฐ ์ ํ๋๋ฉฐ, ๋จ์ ๊ตฌํ์์๋ ์์ฒญ์ด ์์ฐจ ์ฒ๋ฆฌ๋์ด throughput์ด ๊ธ๊ฒฉํ ๋จ์ด์ง๋ค.
-
12:18
๊ธฐ์ด: ํ์ดํ๋ผ์ธ๊ณผ attention ๊ณ์ธต ์ถ๋ก ํ์ดํ๋ผ์ธ์ ํ ํฐํ, ์๋ฒ ๋ฉ, transformer ์ธต, ๋ค์ ํ ํฐ ์์ฑ์ ๋ฐ๋ณต์ผ๋ก ๊ตฌ์ฑ๋๋ฉฐ, ๋๋ถ๋ถ์ ๊ณ์ฐ์ transformer์ ๊ทธ ์์ attention์์ ๋ฐ์ํ๋ฏ๋ก ๋ณ๋ชฉ์ ์์ธ์ attention ๊ด์ ์์ ๋ด์ผ ํ๋ค.
-
14:53
KV cache ์ํ๊ณผ GPU ๋ฉ๋ชจ๋ฆฌ ๋ถ๋ฐฐ ํ ํฐ๋น KV ํฌ๊ธฐ๋ ๋ชจ๋ธ ๊ตฌ์กฐ์ ๋ฐ๋ผ ๊ณ ์ ์ ์ผ๋ก ์ปค์ง๋ฉฐ, ์ปจํ ์คํธ ๊ธธ์ด์ ๋์ ์ฌ์ฉ์ ์๊ฐ ๋์๋ก GPU์ ๋จ์ ๋ฉ๋ชจ๋ฆฌ๊ฐ ๋น ๋ฅด๊ฒ ์์ง๋๋ค. ๋ฐ๋ผ์ model weights์ ๊ณ ์ ์ค๋ฒํค๋๋ฅผ ์ ์ธํ ๋๋จธ์ง ๊ณต๊ฐ์ KV cache๊ฐ ์ด๋ป๊ฒ ์ ์ ํ๋์ง ๊ณ์ฐํด์ผ ํ๋ค.
-
20:41
Prefill๊ณผ decode, ๊ทธ๋ฆฌ๊ณ decode๊ฐ ๋ฉ๋ชจ๋ฆฌ ๋ณ๋ชฉ์ธ ์ด์ prefill์ ์ ๋ ฅ ์ ์ฒด์ ๋ํ KV ์์ฑ๊ณผ attention ๊ณ์ฐ์ ํ ๋ฒ์ ์ํํ๋ ๊ณ์ฐ ์ง์ฝ ๋จ๊ณ๋ผ TTFT๋ฅผ ๊ฒฐ์ ํ๊ณ , decode๋ ๋งค ํ ํฐ๋ง๋ค ์ด์ KV๋ฅผ ๊ณ์ ์ฝ์ด์์ผ ํด์ ๊ณ์ฐ๋ณด๋ค ๋ฉ๋ชจ๋ฆฌ ๋์ญํญ์ ๋ ํฌ๊ฒ ๋ฌถ์ธ๋ค.
-
28:41
์ฒ๋ฆฌ๋, ๋์์ฑ, ๊ทธ๋ฆฌ๊ณ ํธ๋ ์ด๋์คํ ์ผ๊ฐํ ์ปจํ ์คํธ ๊ธธ์ด, ๋์ ์ฌ์ฉ์ ์, ์ง์ฐ์๊ฐ์ ๋์์ ์ต์ ํํ๊ธฐ ์ด๋ ต๊ณ , ํ๋๋ฅผ ํค์ฐ๋ฉด ๋ค๋ฅธ ํ๋๋ฅผ ์ค์ฌ์ผ ํ๋ค. ๋ฐ๋ผ์ ์ฑ๋ด, ๋น๋๊ธฐ agent ์์ ์ฒ๋ผ ์ํฌ๋ก๋ ์ฑ๊ฒฉ์ ๋ฐ๋ผ ํ์งยท์ง์ฐยท์ฒ๋ฆฌ๋์ ์ฐ์ ์์๋ฅผ ๋ค๋ฅด๊ฒ ์ก์์ผ ํ๋ค.
-
34:02
์ฉ๋ ๊ณ์ฐ๊ธฐ์ GPU ์ ํ GPU ์ ํ์ ๋ชฉํ ์ง์ฐ์๊ฐ, ์ต์ ๋์ ๋ฐฐ์น, ํ์ฉ ๊ฐ๋ฅํ ์ปจํ ์คํธ ๊ธธ์ด๋ฅผ ๋จผ์ ๊ณ ์ ํ ๋ค ์ญ์ฐํด์ผ ํ๋ค. ๊ฒ๋ณด๊ธฐ ๋น์ฉ์ด ๋ ๋น์ผ GPU๋ผ๋ ์ค์ ๋ก๋ ๋ ๋ง์ ํ ํฐ์ ์ฒ๋ฆฌํด cost per million tokens๊ฐ ๋ ๋ฎ์์ง ์ ์๋ค.
-
37:10
๋ชจ๋ธ ์ต์ ํ, quantization, attention ๋ณํ ๋ชจ๋ธ ํฌ๊ธฐ๋ FP16, int8, int4, NF4 ๊ฐ์ quantization์ผ๋ก ์ค์ผ ์ ์๊ณ , attention์ multi-head attention์์ grouped query attention, multi-query attention, multi-head latent attention, sparse attention์ผ๋ก ์งํํ๋ฉฐ ๋ฉ๋ชจ๋ฆฌ์ ๊ณ์ฐ๋์ ์ค์ธ๋ค.
-
48:04
Flash attention, ์ ์ํ, ๋ฐ๋ชจ Flash attention์ ํฐ attention ํ๋ ฌ์ ์์ tile๋ก ์ชผ๊ฐ HBM๊ณผ shared memory ์ด๋์ ์ค์ฌ ๋ ํจ์จ์ ์ผ๋ก softmax์ ํ๋ ฌ๊ณฑ์ ์ํํ๋ค. ๋ํ attention ๋ณํ๋ค์ ํ์ง๊ณผ ์์ถ๋ฅ , ๋ณ๋ ฌ์ฑ์ ๊ท ํ์ด ๋ค๋ฅด๋ฉฐ, ์ค์ ๋ฐ๋ชจ์์๋ MLA๊ฐ ์๋นํ ๋ฉ๋ชจ๋ฆฌ ์ ๊ฐ ํจ๊ณผ๋ฅผ ๋ณด์ธ๋ค.
-
1:00:14
์๋น ์ต์ ํ, paging, batching, caching ์๋น ์ธก๋ฉด์์๋ KV cache๋ฅผ ํ์ฉํด ์ค๋ณต ๊ณ์ฐ์ ์ค์ด๊ณ , paging์ผ๋ก ๋ฉ๋ชจ๋ฆฌ ๋จํธํ๋ฅผ ์ํํ๋ฉฐ, continuous batching์ผ๋ก GPU ์ ํด ์๊ฐ์ ์ค์ธ๋ค. prefix caching๊ณผ KV quantization์ ๋ฐ๋ณต ์์ฒญ๊ณผ ์บ์ ํฌ๊ธฐ ๋ฌธ์ ๋ฅผ ์ํํ๋ค.
-
1:06:12
vLLM ๋ฒค์น๋งํน๊ณผ speculative decoding vLLM์ ๊ธฐ๋ณธ์ ์ผ๋ก paging, continuous batching, KV cache๋ฅผ ์ ๊ณตํด ๊ธฐ์ค์ ๋๋น ์ฒ๋ฆฌ๋์ ํฌ๊ฒ ๋์ด๋ฉฐ, prefix caching๊ณผ KV quantization์ ๋ํ๋ฉด ๋ฉ๋ชจ๋ฆฌ ํจ์จ์ด ์ถ๊ฐ๋ก ์ข์์ง๋ค. speculative decoding์ ์์ ๋ชจ๋ธ์ด ๋ฏธ๋ฆฌ ํ ํฐ์ ์ ์ํ๊ณ ํฐ ๋ชจ๋ธ์ด ๊ฒ์ฆํ๋ ๋ฐฉ์์ผ๋ก decode ๊ฐ์์ ๋ ธ๋ฆฐ๋ค.
-
1:17:27
vLLM๊ณผ SGLang ๋น๊ต, ์์ง ์ ํ, ๋ค์ ํ์ต ์ฃผ์ ํ์ค API workload์์๋ vLLM๊ณผ SGLang์ ์ฐจ์ด๊ฐ ํฌ์ง ์์ง๋ง, agentic branching์ฒ๋ผ ๋ฐ๋ณต์ ์ด๊ณ ๊ตฌ์กฐํ๋ ์ํฌํ๋ก์์๋ SGLang์ด ์ ๋ฆฌํ ์ ์๋ค. ์ดํ์๋ ๋ ๊น์ benchmark ์ฝ๊ธฐ, KV eviction ์ ๋ต, hybrid memory, distributed LLM inference๋ฅผ ๊ณต๋ถํ๋ ํ๋ฆ์ด ์ ํจํ๋ค.
์ค๋ฌด ์ธ์ฌ์ดํธ
- GPU ์ ํ์ ์คํ ๋น๊ต๋ณด๋ค ๋จผ์ ๋ชฉํ TTFT, ๋์ ์ฌ์ฉ์ ์, ์ปจํ ์คํธ ๊ธธ์ด๋ฅผ ๊ณ ์ ํ ๋ค KV cache ๊ณ์ฐ์ผ๋ก ์ญ์ฐํด์ผ ํ๋ค.
- ์ถ๋ก ์ต์ ํ๋ ๋ชจ๋ธ ์์ถ๊ณผ ์๋น ์ต์ ํ๋ฅผ ๋ถ๋ฆฌํด์ ๋ณด๋, ์ค์ ๋น์ฉ ์ ๊ฐ์ ๋ ์ถ์ ํจ๊ป ์ ์ฉํ ๋ ๊ฐ์ฅ ํฌ๊ฒ ๋์จ๋ค.
- ํ์ค ์ฑ๋ด๊ณผ agentic workload๋ ๋ณ๋ชฉ์ด ๋ค๋ฅด๋ฏ๋ก, ๋์ผํ inference engine์ ๊ธฐ๋ณธ๊ฐ์ผ๋ก ์ฐ๋๋ผ๋ ์ํฌ๋ก๋๋ณ๋ก vLLM๊ณผ SGLang์ ๊ตฌ๋ถํด ๊ฒํ ํด์ผ ํ๋ค.