Американский стартап Cerebras Systems, который занимается разработкой чипов для машинного обучения и других сложных задач, сообщил о новом мировом рекорде производительности в области молекулярной динамики.
В исследовании участвовали Сандийские национальные лаборатории (SNL), Ливерморская национальная лаборатория имени Лоуренса (LLNL) и Лос-Аламосская национальная лаборатория (LANL), входящие в состав Министерства энергетики США (DOE). Расчёты проводились на системе с использованием специализированных ускорителей Cerebras Wafer Scale Engine (WSE).
Сообщается, что исследователи впервые в истории молекулярной динамики достигли результата более 1 миллиона шагов моделирования в секунду (TPS). В частности, было продемонстрировано значение около 1,1 миллиона TPS на платформе Cerebras CS-2 с чипами Wafer-Scale Engine 2 (WSE-2), содержащими 850 тысяч тензорных ядер и 40 гигабайт памяти SRAM. Для сравнения: у суперкомпьютера экзафлопсного класса Frontier, занимающего второе место в текущем рейтинге TOP500, результат составляет 1470 TPS. Таким образом, система Cerebras обеспечивает 748-кратное увеличение производительности при решении задач молекулярной динамики. Энергопотребление комплекса Cerebras составляет 27 киловатт по сравнению с 21 мегаваттом у Frontier.
Также система Cerebras превосходит Anton 3 — самый мощный в мире суперкомпьютер для молекулярной динамики. Anton 3 использует 512 специализированных процессоров и потребляет 400 киловатт энергии. Его показатель быстродействия составляет 980 тысяч TPS. Таким образом, система Cerebras демонстрирует прирост примерно на 20 %.
Ускорители Cerebras, как ожидается, откроют новые горизонты для исследований в различных областях, таких как создание материалов нового поколения, разработка перспективных лекарств и инновации в области возобновляемой энергетики.
Стоит упомянуть, что ранее Сандийские национальные лаборатории запустили ИИ-систему Kingfisher на чипах Cerebras WSE-3. А компания Cerebras сама создала «самую мощную в мире» ИИ-платформу для инференса.
По материалам:
servernews