Views
No views yet
xlm-roberta-large backbone, so their word-embedding table is stored a single time and injected into the reranker at load — a smaller download at no measured quality loss, with no retraining..embed(...) or .rerank(...).1from modeling_baa import BaaEmbeddingReranker # included in this repo
2
3m = BaaEmbeddingReranker("baa-ai/Merino-XL")
4qv = m.embed(["how does a cross-encoder reranker work?"], is_query=True)[0]
5dv = m.embed(["a cross-encoder scores a (query, document) pair jointly",
6 "bi-encoders embed query and document separately for fast retrieval"])
7ranked = m.rerank("how does a cross-encoder reranker work?",
8 ["a cross-encoder scores a (query, document) pair jointly",
9 "the mitochondria is the powerhouse of the cell"])
10# -> [(doc, score), ...] sorted best-first| Embedding dim | 1024 |
| Parameters | ~880M (embedder + reranker, shared word-embedding table) |
| Languages | multilingual |
| Max sequence length | 512 |
| Hardware | CPU / edge / GPU |
LICENSE.xlm-roberta-large backbone under the MIT License — see LICENSE-xlm-roberta-large.txt.