Why did we open-source our inference engine? Read the post

mixedbread-ai/mxbai-colbert-large-v1

The crispy rerank family from Mixedbread.

Architecture
BERT
Parameters
335M
Tasks
Encode
Outputs
Multi-Vec
Dimensions
Multi-Vec: 128
Max Sequence Length
512 tokens
License
apache-2.0

Benchmarks

CQADupstackPhysicsRetrieval

scientific retrieval en

Duplicate question retrieval from StackExchange Physics

Corpus: 38,314 Queries: 1,039
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 30.0K
Corpus p50 68.7ms
Query TPS 3.3K
Query p50 49.0ms
default
Performance L4 b1 c16
Corpus TPS 32.3K
Corpus p50 65.1ms
Query TPS 3.9K
Query p50 44.7ms
Reference →

CosQA

technology retrieval en

Code search with natural language queries

Corpus: 6,267 Queries: 500
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 12.5K
Corpus p50 62.5ms
Query TPS 2.2K
Query p50 43.6ms
default
Performance L4 b1 c16
Corpus TPS 16.3K
Corpus p50 51.4ms
Query TPS 2.4K
Query p50 40.1ms
Reference →

FiQA2018

finance retrieval en

Financial opinion mining and question answering

Corpus: 57,599 Queries: 648
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 35.0K
Corpus p50 71.8ms
Query TPS 4.2K
Query p50 44.7ms
default
Performance L4 b1 c16
Corpus TPS 38.0K
Corpus p50 66.6ms
Query TPS 4.5K
Query p50 41.8ms
Reference →

LegalBenchConsumerContractsQA

legal retrieval en

Question answering on consumer contracts

Corpus: 153 Queries: 396
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 77.3K
Corpus p50 101.4ms
Query TPS 5.9K
Query p50 45.0ms
default
Performance L4 b1 c16
Corpus TPS 79.7K
Corpus p50 98.6ms
Query TPS 6.3K
Query p50 42.1ms
Reference →

NFCorpus

medical retrieval en

Biomedical literature search from NutritionFacts.org

Corpus: 3,593 Queries: 323
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 51.3K
Corpus p50 92.4ms
Query TPS 1.8K
Query p50 44.1ms
default
Quality
ndcg at 10 0.3467
map at 10 0.1321
mrr at 10 0.5620
Performance L4 b1 c16
Corpus TPS 46.5K
Corpus p50 95.7ms
Query TPS 1.9K
Query p50 42.8ms
Reference →

NanoFiQA2018Retrieval

finance retrieval en

Smaller subset of the FiQA financial QA dataset

Quality
ndcg at 10 0.4833
map at 10 0.4103
mrr at 10 0.5605
Reference →

SCIDOCS

scientific retrieval en

Citation prediction, document classification, and recommendation for scientific papers

Corpus: 25,656 Queries: 1,000
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 36.2K
Corpus p50 81.3ms
Query TPS 3.8K
Query p50 46.5ms
default
Performance L4 b1 c16
Corpus TPS 40.1K
Corpus p50 75.6ms
Query TPS 4.6K
Query p50 39.8ms
Reference →

SciFact

scientific retrieval en

Scientific claim verification using research literature

Corpus: 5,183 Queries: 300
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 46.8K
Corpus p50 89.5ms
Query TPS 5.7K
Query p50 46.2ms
default
Performance L4 b1 c16
Corpus TPS 48.5K
Corpus p50 87.4ms
Query TPS 6.3K
Query p50 41.8ms
Reference →

StackOverflowQA

technology retrieval en

Programming question answering from Stack Overflow

Corpus: 19,931 Queries: 1,994
default_candidates-k-50_candidates-model-Alibaba-NLP__gte-multilingual-base
Performance L4 b1 c16
Corpus TPS 45.2K
Corpus p50 78.9ms
Query TPS 69.9K
Query p50 64.7ms
default
Performance L4 b1 c16
Corpus TPS 49.7K
Corpus p50 74.1ms
Query TPS 76.4K
Query p50 60.7ms
Reference →

Self-hosted inference for search & document processing

Cut API costs by 50x, boost quality with 85+ SOTA models, and keep your data in your own cloud.