A deep learning library with static computational graph and auto-diff in C++.
Switch branches/tags
Nothing to show
Clone or download

README.md

EDXNet

Please go to http://behindthepixels.io/EDXNet/ for more detailed introduction.

EDXNet is a deep learning library independently developed by Edward Liu. It is developed in C++ and only provides C++ interface. It's built with the static computational graph approach like TensorFlow, and implements automatic-differentiation.

A numpy like multi-dimensional tensor library is also included. Template expression is used to make tensor expression evaluation more efficient.

Currently it only supports some of the most common operators like fully connected, convolution, pooling, batchnorm etc. With these simple operators EDXNet can already do many interesting tasks such as training a digit recognition neural net, image classification and even style transfer. The following is one sample result of running style transfer with EDXNet.

More operators and features such as RNN, as well as GPU support is planned in the near future.

The weights for the pre-trained VGG19 network used by EDXNet can be downloaded from here.