janhuenermann/neurojs

A JavaScript deep learning and reinforcement learning library.

JavaScript

4,367

53 commits

updated Oct 10, 2023

See the code

README

neurojs

neurojs

neurojs is a JavaScript framework for deep learning in the browser. It mainly focuses on reinforcement learning, but can be used for any neural network based task. It contains neat demos to visualise these capabilities, for instance a 2D self-driving car.

Feel free to contribute. The development on such projects is more awesome in a community!

Note: this repo is no longer maintained since more general frameworks such as TensorFlow-JS emerged, which I recommend you to use.

Features

  • Implements a full-stack neural-network based machine learning framework
  • Extended reinforcement-learning support
    • Uniform and prioritised replay buffers
    • Advantage-learning (increasing the action-gap) https://arxiv.org/pdf/1512.04860v1.pdf
    • Support for deep-q-networks and actor-critic models (via deep-deterministic-policy-gradients)
  • Binary import and export of network configurations (weights etc.)
  • High-performance

Examples

Running the examples

npm install
npm start

Open http://localhost:8080/examples/ in your browser and select the demo you want to run.

What's next?

  • More examples (pong, pendulum, snake,improved cars, etc.)
  • Support for web workers
  • LSTM and backpropagation through time
deep-learning
javascript
machine-learning
neural-network
reinforcement-learning
self-driving-car

janhuenermann/neurojs

A JavaScript deep learning and reinforcement learning library.

JavaScript

4,367

53 commits

updated Oct 10, 2023

See the code

README

neurojs

neurojs

neurojs is a JavaScript framework for deep learning in the browser. It mainly focuses on reinforcement learning, but can be used for any neural network based task. It contains neat demos to visualise these capabilities, for instance a 2D self-driving car.

Feel free to contribute. The development on such projects is more awesome in a community!

Note: this repo is no longer maintained since more general frameworks such as TensorFlow-JS emerged, which I recommend you to use.

Features

  • Implements a full-stack neural-network based machine learning framework
  • Extended reinforcement-learning support
    • Uniform and prioritised replay buffers
    • Advantage-learning (increasing the action-gap) https://arxiv.org/pdf/1512.04860v1.pdf
    • Support for deep-q-networks and actor-critic models (via deep-deterministic-policy-gradients)
  • Binary import and export of network configurations (weights etc.)
  • High-performance

Examples

Running the examples

npm install
npm start

Open http://localhost:8080/examples/ in your browser and select the demo you want to run.

What's next?

  • More examples (pong, pendulum, snake,improved cars, etc.)
  • Support for web workers
  • LSTM and backpropagation through time
deep-learning
javascript
machine-learning
neural-network
reinforcement-learning
self-driving-car

Significant stargazers

Sebastian "Sebbie" Silbermann

1,838 followers · starred Feb 2017

Anton Bukov

986 followers · starred Dec 2017

John Hurliman

137 followers · starred Mar 2017

Brian Kassouf

142 followers · starred Feb 2017