1 unstable release

0.1.0 Feb 29, 2020

#142 in Machine learning

BSD-3-Clause

200KB
3.5K SLoC

neuro

HomepageAPI DocumentationExamples

Neuro is a deep learning library that runs on the GPU. The library is designed to be very modular and allow users to easily add custom activation functions, loss functions, layers, and optimizers. The library presently supports:

  • Layers: BatchNorm, Conv2D, Dense, Dropout, Flatten, MaxPool2D.
  • Optimizers: Adadelta, Adam, RMSprop, SGD.
  • Activations: LeakyReLU, Linear, ReLU, Sigmoid, Softmax, Tanh.
  • Loss functions: BinaryCrossEntropy, CrossEntropy, MeanAbsoluteError, MeanSquaredError, SoftmaxCrossEntropy.

Additionaly, many initialization schemes are available. The current implementation allows the creation of feedforward and convolutional neural networks. It is planned to add recurrent neural networks in the future.

Installation

The crate is powered by ArrayFire to perform all operations on the GPU. The first step is therefore to install this library. When building a project, the path to the ArrayFire library must be in the path environment variables. For instance for a typical install (on Unix):

export DYLD_LIBRARY_PATH=/opt/arrayfire/lib

The models trained with neuro can be saved in the Hierarchical Data Format (HDF5). In order to do so, HDF5 1.8.4 or newer must be installed. Installation files can be found on the HDF Group website. macOS users can install it with homebrew:

brew install hdf5

To start using the library, add the following line to the project's Cargo.toml file:

[dependencies]
neuro = "0.1.0"

It is highly recommended to build the project in release mode for considerable speedup (e.g. cargo run my_project --release). In order to quickly get started, check out the examples.

Dependencies

~12MB
~213K SLoC