Skip to main content

Module quantized_nn

Module quantized_nn 

Source
Expand description

Utilities for quanitized network layers

This module contains various implementations of standard neural network layers, modules and utilities including embedding, linear layers, and various normalization techniques. Most implementations provide quantized weights support.

Structs§

Embedding
Linear
RmsNorm

Functions§

layer_norm
layer_norm_no_bias
linear
linear_b
linear_no_bias