À propos du projet

TensorFlow.js est une bibliothèque JavaScript open source accélérée par le matériel pour entraîner et déployer des modèles de machine learning. Elle fournit des API bas niveau flexibles pour le calcul numérique et les réseaux de neurones, ainsi qu'une API de couches haut niveau similaire à Keras. La bibliothèque prend en charge plusieurs backends, notamment CPU, WebGL, WASM, WebGPU, Node.js et React Native, permettant aux modèles de s'exécuter dans les navigateurs, Node.js et les environnements mobiles. Les utilisateurs peuvent créer des modèles à partir de zéro, importer des modèles TensorFlow SavedModel ou Keras pré-entraînés à l'aide du convertisseur, et réentraîner des modèles existants avec des données côté client. Le dépôt combine plusieurs packages : Core, Layers, Data, Converter, Vis et AutoML. Il propose également des outils de benchmark pour mesurer les métriques de performance localement ou sur des appareils distants. TensorFlow.js fait partie de l'écosystème TensorFlow et comprend des tutoriels, des exemples et une galerie de projets associés.