Shumai

Shumai — библиотека тензорных вычислений, которая позволяет JavaScript-разработчикам создавать и обучать модели машинного обучения с производительностью промышленного уровня. Она объединяет веб-разработку и численные вычисления, предоставляя дифференцируемые тензоры, автоматическое вычисление градиентов и эффективные матричные операции.

Ключевые возможности

• Быстрое выполнение благодаря FFI-привязкам и среде выполнения Bun
• Поддержка ускорения на GPU для крупномасштабных вычислений
• Автоматическое дифференцирование для обратного распространения ошибки и отслеживания градиентов
• Сетевые возможности для распределённых процессов машинного обучения
• Стандартные операции с тензорами и базовые компоненты нейронных сетей
• Ориентированная на TypeScript архитектура со строгой типизацией

Библиотека предназначена для JavaScript-разработчиков, которым нужны серьёзные возможности машинного обучения без перехода на другой язык. Она особенно полезна для интеграции моделей в веб-приложения, создания конвейеров обработки данных в Node.js и экспериментов с нейронными сетями на языке, уже знакомом большинству фронтенд- и фулстек-разработчиков. Проект пока остаётся экспериментальным, но активно поддерживается, поэтому подходит для разработки и исследований.