Download
GPU accelerated Neural networks in JavaScript for Browsers and Node.js. brain.js is a GPU accelerated library for Neural Networks written in JavaScript. Brain.js depends on a native module headless-go for GPU support. In most cases installing brain.js from npm should just work. However, if you run into problems, this means prebuilt binaries are not able to download from GitHub repositories and you might need to build it yourself. Brain.js is super simple to use. You do not need to know Neural Networks in detail to work with this. Brain.js performs computations using GPU and gracefully fallback to pure JavaScript when GPU is not available. Brain.js provides multiple neural network implementations as different neural nets can be trained to do different things well. Easily export and import trained models using JSON format or as a function. Host pre-trained models on your website easily.
Features
- For Browsers and Node.js
- Simple, fast and easy to use
- Brain.js performs computations using GPU and gracefully fallback to pure JavaScript when GPU is not available
- Brain.js provides multiple neural network implementations as different neural nets can be trained to do different things well
- Easily export and import trained models using JSON format or as a function. Host pre-trained models on your website easily
- The Brain.js implementation is highly modular to allow easy extendability
Project Samples
