package kaun

  1. Overview
  2. Docs

Module Kaun_modelsSource

Kaun_models: Model zoo with pre-defined neural network architectures.

Provides implementations of popular models that can be easily instantiated and used with pretrained weights from HuggingFace or trained from scratch.

Classic Vision Models

Sourcemodule LeNet : sig ... end

LeNet-5: Classic CNN for handwritten digit recognition.

Language Models

Sourcemodule Bert : sig ... end

BERT: Bidirectional Encoder Representations from Transformers.

Sourcemodule GPT2 : sig ... end

GPT-2: Generative Pre-trained Transformer 2 for causal language modeling.