autograph 0.0.1

Machine Learning Library built on ndarray.
Documentation

License Build Status

autograph

Machine Learning Library for Rust

Features

  • Layers
    • Sequential
    • Dense
    • Conv
    • MaxPool
    • Relu
  • Functions
    • Softmax
    • CrossEntropyLoss
    • ClassificationMatches
  • Initializers
    • Zeros
    • Random
    • HeNormal
    • XavierUniform
  • Optimizers
    • SGD

Examples

  • Lenet5

Next Steps

Pull Requests welcome!

  • Multithreaded training / inference
  • Saving / Loading of Model Structure and Parameters