Cerebras пусна нова версия на огромния си процесор за машинно обучение

Американската компания Cerebras Systems представи най-големия и най-ефективен процесор в света, съобщи Tom's Hardware. Новият продукт, наречен Wafer Scale Engine 2 (WSE-2), е наследник на WSE чипа, представен от компанията през 2019 г. Обновената версия се очаква на пазара през третата четвърт на 2021 г.

Размерите на чипа са 220х220 мм - производството на една такава микросхема заема цяла 300 мм силициева пластина. Чипът има площ от 46 255 кв.мм, приблизително 56 пъти по-голяма от тази най-големия графичен процесор, Nvidia A100, и съдържа 2,6 трлн, транзистори.

Процесорът се произвежда по 7 нм технология, съдържа 850 хиляди програмируеми ядра, оптимизирани за изчисления във векторно пространство, а количеството вградена SRAM памет достига 40 GB. Ядрата работят с честота от 2,5 до 3 GHz - същата като тази на чипа от първо поколение. Пропусквателната способност на паметта обаче се е увеличила от 9 на 20 PB/s (петабайта в секунда).

Благодарение на увеличената взаимосвързаност, общата производителност на чипа е нараснала от 100 Pbps на 220 Pbps. Предвидена е възможност за "заобикаляне" на дефектни зони. Това е необходимо, тъй като при толкова впечатляващ брой ядра е много вероятна появата на дефекти по време производството.

Системите с изкуствен интелект разчитат на паралелни изчисления, което изисква многопроцесорни системи. Тяхната производителност до голяма степен е ограничена от ниската скорост на междупроцесорни комуникации, в която участва оперативната памет. Cerebras, със своите чипове от серията WSE, решава този проблем, като поставя огромно количество памет директно в кристала на процесора. В нея се зарежда наведнъж цялата информация, необходима за извършване на изчисления с използване на всички работещи ядра. Благодарение на това натоварването се разпределя равномерно, така че обработката на данните да е завършена едновременно от всички ядра.

Както и преди, процесорът се доставя като част от персонализирана 15U система, наречена CS-2. Миниатюрният суперкомпютър включва всички необходими мрежови компоненти и система за водно охлаждане. Цялостната система консумира около 17 kW електроенергия, което е с 2 kW по-малко от предишния модел. Според представители на Cerebras, изчислителната мощност на CS-2 е еквивалентна на "стотици GPU-базирани сървъри, консумиращи стотици киловати".

Софтуерът, предоставен от компанията, позволява на потребителите на CS-2 да създават свои собствени модели за машинно обучение, използвайки популярни рамки като PyTorch и TensorFlow.

CS-2 е предназначен за използване в центрове за данни за обработка на паралелни изчисления в областта на машинното обучение и изкуствения интелект. Изследователските лаборатории са сред потенциалните клиенти.

Първата му версия се използва Националната лаборатория Аргон (контролирана от Министерството на енергетиката на САЩ) за изследване на ракови заболявания и моделиране на черни дупки. Според Tom's Hardware продуктът може да представлява интерес за военните и разузнавателните агенции.