208-кратное ускорение + прогноз на 5 микросекунд, запуск KMeans на H200 прямо достиг 61% пиковых FLOPS, эта команда из Беркли довела Triton до совершенства

Сайт币界网 сообщает, что открытая библиотека машинного обучения FlashLib, раскрытая OneMillion_AI, разработана командами из Беркли и других, охватывает 15 высокоуровневых операторов, основана на Triton и Cutedsl, и достигает значительного ускорения на GPU H200 для операторов KMeans, KNN и других, до 208 раз.
По сравнению с cuML 25.10, показатели ускорения составляют: KMeans 26x, KNN 19x, HDBSCAN 40x, TruncatedSVD 208x;
максимальная пиковая производительность FLOPS KMeans достигает 61%, пропускная способность KNN — 85,2%.
Кроме того, FlashLib предоставляет API для предсказания производительности за 5 микросекунд, оценивающего время выполнения и расход видеопамяти, код уже открыт на GitHub.