package kaun

  1. Overview
  2. Docs
Flax-inspired neural network library for OCaml

Install 

dune-project
 Dependency

github.com Readme Changelog Edit opam file Versions (3)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

raven-1.0.0.alpha2.tbz
sha256=93abc49d075a1754442ccf495645bc4fdc83e4c66391ec8aca8fa15d2b4f44d2
sha512=5eb958c51f30ae46abded4c96f48d1825f79c7ce03f975f9a6237cdfed0d62c0b4a0774296694def391573d849d1f869919c49008acffca95946b818ad325f6f

doc/kaun.models/Kaun_models/index.html

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.