Skip to main content

Module arch

Module arch 

Source
Expand description

Model architecture trait and variant enum.

The ModelArch trait defines how a model architecture composes Driver primitives into a complete forward pass (embeddings -> encoder layers -> pooling -> normalization).

Each architecture (ClassicBert, ModernBert) is implemented once and works with any driver backend via generics.

Modules§

classic_bert
ClassicBert architecture (BGE-small-en-v1.5).
modern_bert
ModernBERT architecture (nomic-ai/modernbert-embed-base).

Enums§

ArchVariant
Supported model architectures.

Traits§

ModelArch
Model architecture that composes Driver primitives into a forward pass.