Module `Tolk_nn`

module Embedding : sig ... end

Token embedding: a lookup table from integer indices to dense vectors.

module Layer_norm : sig ... end

Layer normalization over the last axis.

module Linear : sig ... end

Affine transformation of the last axis.

module State : sig ... end

Model state: loading tensors from disk and binding them to parameters.