Shumai

Shumai é uma biblioteca de tensores que permite a desenvolvedores JavaScript criar e treinar modelos de machine learning com desempenho adequado para produção. Ela aproxima o desenvolvimento web da computação numérica ao oferecer tensores diferenciáveis, cálculo automático de gradientes e operações matriciais eficientes.

Destaques

• Execução rápida por meio de bindings FFI com o runtime Bun
• Aceleração por GPU para cálculos em grande escala
• Diferenciação automática para retropropagação e acompanhamento de gradientes
• Conectividade de rede para fluxos distribuídos de machine learning
• Operações padrão com tensores e componentes básicos para redes neurais
• Projeto centrado em TypeScript, com tipagem robusta

A biblioteca é voltada a desenvolvedores JavaScript que precisam de recursos avançados de ML sem trocar de linguagem. É particularmente útil para integrar modelos a aplicações web, criar pipelines de processamento de dados em Node.js e experimentar redes neurais em uma linguagem já conhecida pela maioria dos profissionais de front-end e full-stack. O projeto ainda é experimental, mas recebe manutenção ativa, sendo adequado para casos de uso em desenvolvimento e pesquisa.