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 ClassicBertarchitecture (BGE-small-en-v1.5).- modern_
bert ModernBERTarchitecture (nomic-ai/modernbert-embed-base).
Enums§
- Arch
Variant - Supported model architectures.