Module layers

Structs

Dense

Feed-forward function with activiation function.

DenseConfig

Configuration for Dense layer.

Pooling

Performs pooling (max or mean) on the token embeddings.

PoolingConfig

Configuration for Pooling layer.