docbert-pylate 0.9.0

Rust library for late interaction (ColBERT) models, vendored into the docbert workspace.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

pub mod builder;
pub mod error;
pub mod model;
pub mod modernbert;
pub mod pooling;
pub mod types;
pub mod utils;

pub use builder::ColbertBuilder;
pub use error::ColbertError;
pub use model::{BaseModel, ColBERT};
pub use pooling::hierarchical_pooling;
pub use types::{
    EncodeInput,
    EncodeOutput,
    RawSimilarityOutput,
    Similarities,
    SimilarityInput,
};
pub use utils::normalize_l2;