תעשיית ה-AI מזיזה את מוקד ההשקעה מאימון מודלים ענקיים להרצתם היעילה בקנה מידה עסקי, ולשם כך אנבידיה, גוגל ו-AWS מפתחות ארכיטקטורות שבבים הטרוגניות - שבבים ייעודיים לשלבים שונים בתהליך ה-Inference. הגישה מבטיחה חיסכון בעלויות אך מציבה אתגרי תוכנה ורשת חדשים.
✦ נכתב אוטומטית על ידי AIמבוסס על מקורות גלויים · עבר בקרת איכות אוטומטית
בשנים האחרונות המירוץ בתעשיית ה-AI התנהל סביב שאלה אחת: מי יבנה את המודל הגדול ביותר, הכי מהר. עכשיו מרכז הכובד זז. מודל מאמנים פעם אחת, ומשרתים באמצעותו מיליארדי בקשות - וככל שצ'אטבוטים, סוכני קוד ומערכות אנליטיקה נכנסים לשימוש עסקי רחב, ההשקעה עוברת מאימון (Training) להיסק (Inference), כך על פי דיווח בגיקטיים. מודלי Reasoning וסוכני AI אוטונומיים רק מאיצים את המגמה, שכן כל משימה מייצרת שרשרת ארוכה של קריאות למודל וצריכת הטוקנים גדלה בהתאם.
שבב אחד לא מספיק
לפי המקור, תהליך ה-Inference אינו פעולה אחת אלא שרשרת שלבים שונים מהותית זה מזה. בשלב ה-Prefill המודל "קורא" את הקלט ובונה הקשר - משימה עתירת חישוב ומקבילית. בשלב ה-Decode נוצרים טוקני התשובה אחד אחרי השני - תהליך סדרתי, צורך זיכרון גבוה ורגיש להשהיה. בנוסף יש את שלבי הטוקניזציה וניהול ה-KV Cache, ההקשר שהמודל צובר לאורך שיחה. שבב יחיד שמנסה להצטיין בכל השלבים, כך לפי הדיווח, "משלם מחיר": זיכרון HBM יקר שיושב מובטל בשלב אחד, או כוח חישוב שממתין לזיכרון בשלב אחר.
מכאן ההיגיון מאחורי הארכיטקטורה ההטרוגנית - שבבים נפרדים לכל שלב. AWS מפתחת כבר שנים שני קווי שבבים נפרדים: Trainium לאימון ו-Inferentia להיסק. גוגל פיצלה את דור ה-TPU השמיני שלה לשני מוצרים - TPU 8t לאימון ו-TPU 8i להיסק. אנבידיה, מצדה, חשפה LPU ראשון המיועד להיסק ומבוסס על קניין רוחני של Groq, שאותה רכשה, לצד ה-GPU החדש שלה מדגם Rubin. דוגמה נוספת שמצוינת בדיווח היא שיתוף הפעולה שהוכרז השנה בין AWS ל-Cerebras: שבבי Trainium מטפלים בשלב ה-Prefill ומערכות Cerebras מטפלות ב-Decode, כשכל בקשה עוברת בין מעבדים של יצרנים שונים תוך כדי עיבוד יחיד.
המחיר: תוכנה ורשת מורכבות יותר
לגישה ההטרוגנית יש עלות נלווית. האתגר הראשון הוא שכבת התוכנה: כל משפחת מאיצים "מדברת" בשפה משלה - CUDA אצל אנבידיה, Neuron אצל AWS, ושפות ייעודיות אצל שחקנים אחרים. הרצת מודל אחד על תשתית מגוונת דורשת שכבת הפשטה שמתרגמת בין הסביבות, ומעליה מנגנון תזמון שמקצה בזמן אמת את המעבד המתאים לכל שלב - טעות בהקצאה, לפי הדיווח, עלולה לבטל את כל היתרון הכלכלי.
האתגר השני הוא הרשת. בקלאסטר הומוגני דפוסי התעבורה צפויים יחסית, אך בסביבה הטרוגנית התעבורה הופכת בלתי אחידה: העברת ה-KV Cache בין מעבדי ה-Prefill למעבדי ה-Decode חייבת להתבצע במהירות גבוהה, ופערי קצב בין סוגי המעבדים יוצרים פרצי תעבורה. מאחר שתהליכי AI הם סינכרוניים במהותם - המערכת כולה מתקדמת בקצב הרכיב האיטי ביותר - כל תקלה ברשת מתורגמת ישירות למעבדים יקרים שעומדים מובטלים.
למה זה חשוב
המשמעות המעשית, כפי שעולה מהדיווח, היא שמדד ה-TCO (עלות הבעלות הכוללת) המסורתי, המבוסס על מספר מעבדים או הספק חישוב גולמי, עלול להטעות בעולם ההיסק: שני קלאסטרים עם אותו מספר מעבדים יכולים להפיק כמות שונה בתכלית של טוקנים בפועל, בהתאם לביצועי הרשת ולהתאמת המעבדים לעומס. ככל הנראה, זו הסיבה שהתעשייה עוברת בהדרגה למדדים הממוקדים בתפוקה בפועל, כמו עלות למיליון טוקנים, במקום במפרט חומרה גולמי - שינוי שמשקף את המעבר הרחב יותר מ"מרוץ בניית המודלים" ל"מרוץ הפעלתם" בקנה מידה כלכלי.