Crate zyx_nn

nn modules for zyx ML library

Modules

prelude

Prelude exports all initialization traits. These are needed to provide module initialization functions for devices.

Structs

BatchNorm

Batch norm

LayerNorm

Lyaer norm layer

Linear

Linear layer