Skip to content

A deep learning library with static computational graph and auto-diff in C++.

Notifications You must be signed in to change notification settings

behindthepixels/EDXNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

Releases

No releases published

Packages

No packages published