تحلیل دقیق تر:
قضیه از این قراره که Cerebras با سیستمهاش داره رکوردهای سرعت Inference رو برای مدلهای غولپیکر جابجا میکنه. چندتا عدد واقعی از بنچمارکهای اخیر:
Llama 3.1 405B: حدود ۹۷۰ توکن بر ثانیه
Qwen3-Coder 480B: حدود ۲۰۰۰ توکن بر ثانیه
GPT-OSS 120B: حدود ۳۰۰۰ توکن بر ثانیه
این اعداد نشون میده معماری Wafer-Scale دیگه یه پروژه تحقیقاتی جالب نیست، بلکه یه راه حل عملی برای رسیدن به سرعتهای خیلی بالاست.
چرا Cerebras اینقدر سریعه؟ معماری Wafer-Scale
برخلاف رویکرد سنتی که توش چندین GPU رو با شبکه به هم وصل میکنیم، Cerebras یه تراشه غولپیکر به اندازه کل یک ویفر سیلیکونی ساخته. جدیدترین مدلش، WSE-3، حدود ۹۰۰ هزار هسته پردازشی بهینه شده برای هوش مصنوعی و ۴۴ گیگابایت حافظه SRAM فوق سریع رو روی یک چیپ یکپارچه کرده.
این معماری چندتا گلوگاه اصلی که تو کلاسترهای GPU وجود داره رو حذف میکنه:
۱. پهنای باند حافظه فضایی: بزرگترین مزیت Cerebras، حافظه On-Chip یا همون SRAM هست. WSE-2 پهنای باند حافظه داخلی حدود 20 PB/s (پتابایت بر ثانیه) داره. این عدد رو مقایسه کنید با پهنای باند حافظه HBM یه GPU مثل H100 که حدود 3 TB/s (ترابایت بر ثانیه) هست. یعنی بیش از ۷۰۰۰ برابر پهنای باند بیشتر. وقتی پارامترهای مدل و KV Cache داخل همین حافظه فوق سریع قرار میگیرن، دیگه نیازی به رفتوآمد به حافظه کندترِ خارج از چیپ نیست و عملیاتهای Memory-Bound مثل Attention به شدت سریع میشن.
۲. حذف گلوگاه شبکه: تو کلاسترهای GPU، مدلهای بزرگ باید بین چندین چیپ Shard بشن. این یعنی بخش زیادی از زمان صرف ارتباط بین GPUها از طریق شبکههایی مثل NVLink یا InfiniBand میشه. توی Cerebras، چون همه هستهها روی یک ویفر هستن، با یه Fabric یا شبکه داخلی 2D Mesh با پهنای باند 220 Pb/s (پتابیت بر ثانیه) به هم وصلن. این یعنی ارتباط بین هستهها هزاران برابر سریعتر از ارتباط بین GPUهاست و عملاً سربار ارتباطی حذف میشه.
۳. پشتیبانی سختافزاری از Sparsity: هستههای پردازشی Cerebras که بهشون SLAC میگن، طوری طراحی شدن که عملیات ضرب در صفر رو به صورت سختافزاری نادیده میگیرن. از اونجایی که وزنها و Activationهای مدلهای زبانی میتونن تا ۹۰٪ اسپارس باشن، این قابلیت بدون کاهش دقت، حجم محاسبات رو ۲ تا ۳ برابر کم میکنه.
۴. تکنیک Weight Streaming: برای مدلهایی که بزرگتر از حافظه SRAM هستن (مثلاً مدلهای 400B)، پارامترها از حافظه خارجی (تا ۱.۲ پتابایت DRAM) به داخل چیپ استریم میشن. این کار با سختافزار اختصاصی مدیریت میشه تا latency به حداقل برسه و نیاز به Sharding سنتی از بین بره.
چرا کلاسترهای GPU کندتر عمل میکنن؟
مشکل اصلی GPUها برای Inference مدلهای بزرگ، معماری توزیعشدهشونه. وقتی یه مدل مثل Llama 3 405B رو روی کلاستر H100 اجرا میکنید، مجبورید مدل رو بین دهها GPU تقسیم کنید. هر توکن که تولید میشه، نیاز به ارتباط all-reduce بین همه GPUها برای محاسبات Attention داره. اینجاست که سرعت کلاستر توسط پهنای باند و latency شبکه محدود میشه، نه قدرت محاسباتی خود GPUها. حتی با سریعترین شبکهها، این ارتباطات سربار قابل توجهی ایجاد میکنن که Cerebras با معماری یکپارچهاش اون رو حذف کرده.
Wafer-scale Computing: Advancements, Challenges, and Future Perspectives
🛠 Join @LLMEngineers Community
Post #156
1.07K
AI Engineers سربراس (Cerebras) یه رویکرد کاملاً متفاوت برای اجرای مدلهای زبانی بزرگ داره که اساساً با معماری GPU-محور انویدیا فرق میکنه. کاربرد اصلیش، حذف کردن گلوگاههاییه که موقع ترین و اینفرنس مدلهای خیلی بزرگ روی کلاسترهای GPU به وجود میاد؛ یعنی مشکلاتی مثل پهنای…
- 🔥 7
- 👍 2
- ❤ 1