meuron 0.4.0

Meuron is a modular neural network library written in rust for training simple neural networks.
Documentation
1
2
3
4
5
6
7
8
pub const BINOP: &str = include_str!("binop.wgsl");
pub const SCALAR: &str = include_str!("scalar.wgsl");
pub const UNARY: &str = include_str!("unary.wgsl");
pub const MATMUL: &str = include_str!("matmul.wgsl");
pub const SOFTMAX: &str = include_str!("softmax.wgsl");
pub const SOFTMAX_VJP: &str = include_str!("softmax_vjp.wgsl");
pub const BROADCAST_ADD: &str = include_str!("broadcast_add.wgsl");
pub const TRANSPOSE: &str = include_str!("transpose.wgsl");