Expand description
Ferrite - A lightweight deep learning framework
This crate provides a dynamic computation graph with automatic differentiation, designed for building and training neural networks.
Modules§
- Layer
- Loss
- Optimizer
- activation
- arithmetic
- blas
- grad
- grad_fn
- loss
- module
- optimizer
- prelude
- reduction
- scalar
- transform
Macros§
Structs§
- AbsGrad
- AddF32
Grad - AddGrad
- Binary
Step Grad - CpuStorage
- DivF32
Grad - DivGrad
- EluGrad
- Leaky
Relu Grad - MatMul
Grad - Mean
Grad - MulF32
Grad - MulGrad
- Parametric
Relu Grad - Permute
Grad - PowF32
Grad - Product
Grad - Relu
Grad - Sigmoid
Grad - Softmax
Grad - SubF32
Grad - SubGrad
- SumGrad
- Swish
Grad - Tanh
Grad - Tensor
Enums§
Constants§
Traits§
- Activation
Ops - Arithmetic
Ops - BlasOps
- Device
Storage - Device
Storage Creation - Device
Storage Static - Display
- Gradient
Function - Loss
Trait - Optimizer
Trait - Reduction
Ops - Transform
Ops