Skip to content

First release

Compare
Choose a tag to compare
@samsja samsja released this 27 Sep 13:17
· 26 commits to master since this release

This is still a prototype of a deep learning library.

What it is so far is an autograd library for float type. It creates dynamic graph via reference counting option of rust and the interior mutability pattern of rust.

What we can do so far is to compose any kind of the following operation:

  • Add
  • Sub
  • Mul
  • Div
  • Relu
    and do a backward pass to get the gradient.

TODO for the next release (hopefully)

  • Use ndarray as a backend to extend it to array composition.
  • implement a full MLP
  • implement some basic optim algo (SGD)
  • Plug everything and train a simple network on a simple task !