package kaun

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

Install

dune-project
 Dependency

Authors

Maintainers

Sources

raven-1.0.0.alpha1.tbz
sha256=8e277ed56615d388bc69c4333e43d1acd112b5f2d5d352e2453aef223ff59867
sha512=369eda6df6b84b08f92c8957954d107058fb8d3d8374082e074b56f3a139351b3ae6e3a99f2d4a4a2930dd950fd609593467e502368a13ad6217b571382da28c

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.