Skalowanie doświadczeń AI dzięki chipom MTIA firmy Meta
Każdego dnia miliardy ludzi na różnorodnych platformach Meta korzystają z niezliczonych funkcji opartych na AI, od spersonalizowanych rekomendacji treści po zaawansowanych asystentów AI. Podstawowe wyzwanie dla Meta, a właściwie dla całej branży, polega na wdrażaniu i ciągłym ulepszaniu tych złożonych modeli AI na skalę globalną, przy jednoczesnym utrzymaniu optymalnej efektywności kosztowej. To wymagające zadanie infrastrukturalne jest realizowane dzięki strategicznej inwestycji Meta w elastyczne, ciągle ewoluujące rozwiązania, których centralnym elementem są ich specjalnie zaprojektowane chipy AI: rodzina Meta Training and Inference Accelerator (MTIA).
Chociaż Meta stawia na zróżnicowane portfolio krzemowe, które wykorzystuje zarówno rozwiązania wewnętrzne, jak i zewnętrzne, chipy MTIA, opracowane w ścisłym partnerstwie z Broadcom, stanowią niezastąpiony element strategii infrastruktury AI firmy Meta. Te opracowane we własnym zakresie akceleratory są kluczowe dla efektywnego kosztowo zasilania doświadczeń AI docierających do miliardów użytkowników, stale dostosowując się do szybko ewoluującego krajobrazu modeli AI.
Iteracyjna ewolucja chipów MTIA firmy Meta
Krajobraz modeli AI znajduje się w stanie ciągłej zmiany, ewoluując w tempie, które często przewyższa tradycyjne cykle rozwoju chipów. Uznając, że projekty chipów oparte na przewidywanych obciążeniach mogą stać się przestarzałe zanim sprzęt trafi do produkcji, Meta przyjęła innowacyjną "strategię prędkości" dla MTIA. Zamiast długich, spekulatywnych okresów rozwoju, Meta stosuje iteracyjne podejście, w którym każda generacja MTIA opiera się na poprzedniej. Obejmuje to wykorzystanie modułowych chipletów, włączanie najnowszych spostrzeżeń dotyczących obciążeń AI i wdrażanie nowych technologii sprzętowych w znacznie krótszym cyklu. Ta ściślejsza pętla sprzężenia zwrotnego zapewnia, że niestandardowy krzem Meta pozostaje ściśle zgodny z dynamicznymi wymaganiami modeli AI, sprzyjając szybszemu wdrażaniu nowych osiągnięć.
Meta już szczegółowo opisała dwie pierwsze generacje, MTIA 100 i MTIA 200, w artykułach naukowych. Bazując na tych podstawach, Meta przyspieszyła rozwój, wprowadzając cztery nowe kolejne generacje: MTIA 300, 400, 450 i 500. Chipy te są już w produkcji lub są przewidziane do masowego wdrożenia w 2026 i 2027 roku. Ta szybka sukcesja pozwoliła Meta na znaczne rozszerzenie zakresu obciążeń MTIA, przechodząc od początkowej inferencji rankingowej i rekomendacyjnej (R&R) do treningu R&R, ogólnych obciążeń Generatywnej AI (GenAI) oraz wysoce zoptymalizowanej inferencji GenAI.
MTIA 300: Kładzenie podwalin pod obciążenia AI
MTIA 300 stanowił kluczowy krok w podróży Meta w kierunku niestandardowego krzemu. Początkowo zoptymalizowany dla modeli R&R, które były dominującymi obciążeniami Meta przed boomem GenAI, jego architektoniczne bloki konstrukcyjne stworzyły solidne podstawy dla kolejnych chipów. Kluczowe cechy wyróżniające MTIA 300 obejmują zintegrowane chiplety NIC, dedykowane silniki wiadomości do odciążania kolektywów komunikacyjnych oraz możliwości obliczeniowe blisko pamięci, zaprojektowane dla kolektywów opartych na redukcji. Te niskoprzewodowe, wysokoprzepustowe komponenty komunikacyjne okazały się kluczowe w umożliwieniu efektywnej inferencji i treningu GenAI w kolejnych generacjach.
MTIA 300 składa się z jednego chipleta obliczeniowego, dwóch chipletów sieciowych i kilku stosów pamięci o wysokiej przepustowości (HBM). Każdy chiplet obliczeniowy zawiera siatkę elementów przetwarzających (PE), strategicznie zaprojektowanych z nadmiarowymi PE w celu zwiększenia wydajności. Każdy PE to zaawansowana jednostka zawierająca dwa wektorowe rdzenie RISC-V, silnik iloczynu skalarnego do mnożenia macierzy, specjalną jednostkę funkcyjną do aktywacji i operacji elementarnych, silnik redukcji do akumulacji i komunikacji między PE oraz silnik DMA do efektywnego przenoszenia danych w lokalnej pamięci podręcznej. Ten skomplikowany projekt podkreślił zaangażowanie Meta w tworzenie wysoce wydajnego i ekonomicznego rozwiązania dla swoich podstawowych zadań AI.
MTIA 400: Osiąganie konkurencyjnej wydajności GenAI
Wraz z bezprecedensowym wzrostem popularności Generatywnej AI, Meta szybko przekształciła MTIA 300 w MTIA 400, aby zapewnić solidne wsparcie dla obciążeń GenAI obok istniejących możliwości R&R. MTIA 400 stanowi znaczący skok, oferując o 400% wyższe FLOPS FP8 i 51% wzrost przepustowości HBM w porównaniu do poprzednika. Podczas gdy MTIA 300 skupiał się na efektywności kosztowej, MTIA 400 został zaprojektowany, aby dostarczać surową wydajność konkurencyjną z wiodącymi komercyjnymi akceleratorami AI.
Osiąga to poprzez połączenie dwóch chipletów obliczeniowych, aby efektywnie podwoić gęstość obliczeniową, oraz poprzez wspieranie ulepszonych wersji MX8 i MX4, kluczowych formatów o niskiej precyzji dla efektywnej inferencji GenAI. Pojedyncza szafa wyposażona w 72 urządzenia MTIA 400, połączone za pośrednictwem przełączanej płyty bazowej, tworzy potężną domenę skalowalności. Systemy te są wspierane przez zaawansowane szafy z chłodzeniem cieczowym wspomaganym powietrzem (AALC), ułatwiając szybkie wdrożenie nawet w starszych centrach danych, pokazując praktyczne podejście Meta do globalnego skalowania swojej infrastruktury AI.
MTIA 450 i 500: Specjalizacja w inferencji GenAI
Przewidując dalszy wykładniczy wzrost zapotrzebowania na inferencję GenAI, Meta dalej udoskoniliła MTIA 400, co doprowadziło do opracowania MTIA 450, a następnie MTIA 500. Te generacje są specjalnie zoptymalizowane pod kątem unikalnych wyzwań inferencji GenAI, koncentrując się na kluczowych postępach w zakresie pamięci i obliczeń.
MTIA 450 poczynił znaczące postępy poprzez:
- Podwojenie przepustowości HBM w porównaniu z poprzednią wersją, co jest kluczowe dla przyspieszenia fazy dekodowania w modelach GenAI.
- Zwiększenie FLOPS MX4 o 75%, przyspieszając obliczenia sieci neuronowej z przeplotem (FFN) typu mixture-of-experts (MoE) powszechne w dużych modelach językowych.
- Wprowadzenie akceleracji sprzętowej w celu zwiększenia efektywności obliczeń uwagi i FFN, łagodząc wąskie gardła związane z Softmax i FlashAttention.
- Innowacje w typach danych o niskiej precyzji, wykraczające poza FP8/MX8, aby zapewnić 6-krotnie większe FLOPS MX4 niż FP16/BF16, z niestandardowymi innowacjami w typach danych, które zachowują jakość modelu i zwiększają FLOPS przy minimalnym wpływie na powierzchnię chipa.
MTIA 500, bazując na sukcesie 450, jeszcze bardziej zwiększył przepustowość HBM o dodatkowe 50% i wprowadził więcej innowacji w typach danych o niskiej precyzji, wzmacniając zaangażowanie Meta w przesuwanie granic wydajności inferencji GenAI. Ten nieustanny dążenie do ulepszeń zapewnia, że doświadczenia AI firmy Meta pozostają na najwyższym poziomie.
Skumulowane postępy we wszystkich tych generacjach są wyraźne. Od MTIA 300 do MTIA 500, przepustowość HBM wzrosła imponująco 4,5-krotnie, podczas gdy FLOPS obliczeniowe odnotowały zdumiewający 25-krotny wzrost (z MX8 MTIA 300 do MX4 MTIA 500). To szybkie przyspieszenie w ciągu dwóch lat jest świadectwem strategii prędkości Meta i jej zdolności do ciągłego ulepszania niestandardowego krzemu. Ta ewolucja jest kluczowa dla operacjonalizacji agentowej sztucznej inteligencji i innych złożonych modeli na dużą skalę.
Oto zestawienie kluczowych specyfikacji rodziny MTIA:
| Cecha | MTIA 300 | MTIA 400 | MTIA 450 | MTIA 500 |
|---|---|---|---|---|
| Matryca Obliczeniowa | 1 | 2 | 2 | 2 |
| Stosy HBM | 4 | 4 | 8 | 8 |
| Przepustowość HBM (GB/s)* | 100 | 151 | 302 | 453 |
| MX8 FLOPS (TFLOPS) | 100 | 400 | 400 | 400 |
| MX4 FLOPS (TFLOPS) | N/A | 200 | 350 | 500 |
| Rozmiar Domeny Skalowania | 18 urządzeń** | 72 urządzenia | 72 urządzenia | 72 urządzenia |
| Kluczowa Optymalizacja | Trening R&R, komunikacja o niskim opóźnieniu | Ogólna GenAI, konkurencyjna surowa wydajność | Inferencja GenAI, HBM, niestandardowa niska precyzja | Inferencja GenAI, HBM, niestandardowa niska precyzja |
*Niektórzy dostawcy podają przepustowość dwukierunkową. Pomnóż wartość w tabeli przez dwa, aby uzyskać odpowiadającą jej przepustowość dwukierunkową. **MTIA 300 jest skonfigurowany z siecią scale-out o wyższej przepustowości (200 GB/s) ze względu na stosunkowo mały rozmiar domeny skalowania i docelowe obciążenia R&R.
Te specyfikacje podkreślają dramatyczne ulepszenia w przepustowości pamięci i mocy obliczeniowej, pokazując, jak każda generacja MTIA jest skrupulatnie projektowana, aby sprostać najbardziej palącym wymaganiom obecnych i przyszłych zastosowań AI, w szczególności zasobochłonnych modeli GenAI.
Nieustanne dążenie Meta do niestandardowych rozwiązań krzemowych poprzez rodzinę MTIA podkreśla jej zaangażowanie w dostarczanie najnowocześniejszych doświadczeń AI miliardom użytkowników na całym świecie. Łącząc wewnętrzne innowacje ze strategicznymi partnerstwami, Meta wciąż na nowo definiuje możliwości skalowalnej i efektywnej kosztowo infrastruktury AI.
Źródło oryginalne
https://ai.meta.com/blog/meta-mtia-scale-ai-chips-for-billions/Często zadawane pytania
What are Meta MTIA chips and what is their purpose?
How many generations of MTIA chips has Meta developed in recent years?
What is Meta's 'velocity strategy' for AI chip development?
How do the newer MTIA chips (400, 450, 500) support Generative AI workloads?
What are the key performance advancements from MTIA 300 to MTIA 500?
Why is High-Bandwidth Memory (HBM) important for GenAI inference performance?
Bądź na bieżąco
Otrzymuj najnowsze wiadomości o AI na swoją skrzynkę.
