Re-exports§
pub use self::kfac::KFACConfig;pub use self::kfac::KFACLayerState;pub use self::kfac::KFACStats;pub use self::kfac::LayerInfo;pub use self::kfac::LayerType;pub use self::kfac::KFAC;pub use self::newton_cg::NewtonCG;
Modules§
- hessian_
approximation - Approximated Hessian computation methods
- kfac
- newton_
cg
Structs§
Enums§
- Hessian
Info - Hessian information for second-order methods
Traits§
- Second
Order Optimizer - Trait for second-order optimization methods