mushin 0.5.0

Computational graphs with reverse automatic differentation in the GPU
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! This module exposes tooling for Deep Learning built upon the **Mushin** auto-grad core.
//!
//! ## Usage
//! ```rust
//! use mushin as mu;
//! use mu::nn::{layers::Linear, activations::relu, losses::mse, optimizers::SGD};
//!
//! let x = mu::eye::<16, 1, 1, 3>(1.0).freeze();
//! let y = mu::eye::<16, 1, 1, 5>(3.0).freeze();
//!
//! let linear = Linear::<16, 3, 5, _, _>::new();
//! let optim = SGD::new(&linear.parameters(), 0.01);
//!
//! for _ in 0..5 {
//!     let z = relu(&linear.forward(&x));
//!     let loss = mse(&z, &y);
//!
//!     loss.backward();
//!     optim.step();
//!     loss.reset();
//! }
//! ```

pub mod activations;
pub mod layers;
pub mod losses;
pub mod optimizers;