Google опубликовала библиотеку Deeplearn.js для машинного обучения в браузере

Google открыла доступ к библиотеке для машинного обучения Deeplearn.js, которая работает в браузере. Пока что она функционирует только в настольной версии Google Chrome, но разработчики работают над поддержкой других устройств и браузеров.

Deeplearn.js даёт возможность обучать нейронные сети в браузере, избегая установки дополнительного софта или бэкенда. Исследователи отметили:

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

Используя WebGL JavaScript API для работы с 2D- и 3D-графикой, Deeplearn.js может производить вычисления на графическом процессоре. Это серьёзно повысит производительность и снизит негативный эффект от сравнительно невысокой скорости JavaScript.

Deeplearn.js имитирует структуру библиотеки TensorFlow и пакета для научных вычислений NumPy:

Мы реализовали некоторые из наиболее популярных операций TensorFlow. В релизной версии библиотеки появится возможность экспортировать веса из контрольных точек TensorFlow и встроить их в веб-страницы для работы с Deeplearn.js.

Хотя библиотека писалась для работы с TypeScript, её можно использовать и с чистым JavaScript. С демонстрациями работы Deeplearn.js можно ознакомиться на её официальной страничке. Среди подобных проектов, которые переносят машинное обучение в браузер, стоит отметить TensorFire, позволяющий запускать нейронные сети в браузере, и ML.js, предоставляющий инструменты для машинного обучения и численного анализа для Node.js.