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