Model reference · open weights
mxbai-embed-large is an open-weight embedding model from mixedbread-ai, listed in the AxForge catalogue. AxForge can bring it up on EU-owned hardware for you on request — with the licence handled where one is required.
About
mixedbread-ai/mxbai-embed-large-v1 Here, we provide several ways to produce sentence embeddings. Please note that you have to provide the prompt Represent this sentence for searching relevant passages: for query if you want to use it for retrieval. Besides that you don't need any prompt. Our model also supports Matryoshka Representation Learning and binary quantization. Quickstart Here, we provide several ways to produce sentence embeddings. Please note that you have to provide the prompt Represent this sentence for searching relevant passages: for query if you want to use it for retrieval. Besides that you don't need any prompt. sentence-transformers Transformers Transformers.js If you haven't already, you can install the Transformers.js JavaScript library from NPM using: You can then use the model to compute embeddings like this: Using API You can use the model via our API as follows: The API comes with native int8 and binary quantization support! Check out the docs for more information. Infinity Evaluation As of March 2024, our model archives SOTA performance for Bert-large sized models on the MTEB. It ourperforms commercial models like OpenAIs text-embedding-3-large and matches the performance of model 20x it's size like the echo-mistral-7b. Our model was trained with no overlap of the MTEB data, which indicates that our model generalizes well across several domains, tasks and text length. We know there are some limitations with this model, which will be fixed in v2. Please find more information in our blog post. Matryoshka and Binary Quantization Embeddings in their commonly used form (float arrays) have a high memory footprint when used at scale. Two approaches to solve this problem are Matryoshka Representation Learning (MRL) and (Binary) Quantization. While MRL reduces the number of dimensions of an embedding, binary quantization transforms the value of each dimension from a float32 into a lower precision (int8 or even binary). <b The model supports both approaches! </b You can also take it one step further, and combine both MRL and quantization. This combination of binary quantization and MRL allows you to reduce the memory usage of your embeddings sign
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | mixedbread-ai |
|---|---|
| Type | Embedding models |
| Parameters (lead) | 335M |
| Context | 512 tokens |
| Variants | 1 |
| Runs with | sentence-transformers |
| Released | 2024-03-07 |
| Popularity | 3.1M downloads / month |
| Likes | 823 |
| Licence | Open weights |
How it works
Variants
Open weights ship in several sizes and precisions. One page, all the variants — pick the one that fits your GPU. VRAM figures are estimates from model size.
| Variant | Params | Precision | VRAM | Fits 16 GB | Weights |
|---|---|---|---|---|---|
| mxbai-embed-large-v1 | 335M | BF16 | ~0.8 GB | ✓ | Weights ↗ |
Benchmarks
As published on the model card — the maker's own numbers, not measured by AxForge.
| Task | Dataset | Metric | Score |
|---|---|---|---|
| Classification | MTEB AmazonCounterfactualClassification (en) | accuracy | 75.045 |
| Classification | MTEB AmazonCounterfactualClassification (en) | ap | 37.736 |
| Classification | MTEB AmazonCounterfactualClassification (en) | f1 | 68.927 |
| Classification | MTEB AmazonPolarityClassification | accuracy | 93.84 |
| Classification | MTEB AmazonPolarityClassification | ap | 90.932 |
| Classification | MTEB AmazonPolarityClassification | f1 | 93.83 |
| Classification | MTEB AmazonReviewsClassification (en) | accuracy | 49.184 |
| Classification | MTEB AmazonReviewsClassification (en) | f1 | 48.742 |
| Retrieval | MTEB ArguAna | map_at_1 | 41.252 |
| Retrieval | MTEB ArguAna | map_at_10 | 57.778 |
| Retrieval | MTEB ArguAna | map_at_100 | 58.233 |
| Retrieval | MTEB ArguAna | map_at_1000 | 58.237 |
| Retrieval | MTEB ArguAna | map_at_3 | 53.45 |
| Retrieval | MTEB ArguAna | map_at_5 | 56.376 |
| Retrieval | MTEB ArguAna | mrr_at_1 | 41.679 |
| Retrieval | MTEB ArguAna | mrr_at_10 | 57.927 |
| Retrieval | MTEB ArguAna | mrr_at_100 | 58.389 |
| Retrieval | MTEB ArguAna | mrr_at_1000 | 58.392 |
| Retrieval | MTEB ArguAna | mrr_at_3 | 53.651 |
| Retrieval | MTEB ArguAna | mrr_at_5 | 56.521 |
| Retrieval | MTEB ArguAna | ndcg_at_1 | 41.252 |
| Retrieval | MTEB ArguAna | ndcg_at_10 | 66.018 |
| Retrieval | MTEB ArguAna | ndcg_at_100 | 67.774 |
| Retrieval | MTEB ArguAna | ndcg_at_1000 | 67.844 |
Using it via the API
Once AxForge deploys mxbai-embed-large for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (mxbai-embed-large below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/embeddings \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"mxbai-embed-large","input":"text to embed"}'
Licence
Open weights under apache-2.0 — commercial use is permitted. Deploy it on AxForge EU hardware on request. Read the licence ↗
Explore