Model reference · open weights
jina-embeddings-small-en is an open-weight embedding model from jinaai, 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
Quick Start The easiest way to starting using jina-embeddings-v2-small-en is to use Jina AI's Embedding API. Intended Usage & Model Info jina-embeddings-v2-small-en is an English, monolingual embedding model supporting 8192 sequence length. It is based on a BERT architecture (JinaBERT) that supports the symmetric bidirectional variant of ALiBi to allow longer sequence length. The backbone jina-bert-v2-small-en is pretrained on the C4 dataset. The model is further trained on Jina AI's collection of more than 400 millions of sentence pairs and hard negatives. These pairs were obtained from various domains and were carefully selected through a thorough cleaning process. The embedding model was trained using 512 sequence length, but extrapolates to 8k sequence length (or even longer) thanks to ALiBi. This makes our model useful for a range of use cases, especially when processing long documents is needed, including long document retrieval, semantic textual similarity, text reranking, recommendation, RAG and LLM-based generative search, etc. This model has 33 million parameters, which enables lightning-fast and memory efficient inference, while still delivering impressive performance. Additionally, we provide the following embedding models: - jina-embeddings-v2-small-en: 33 million parameters (you are here). - jina-embeddings-v2-base-en: 137 million parameters. - jina-embeddings-v2-base-zh: 161 million parameters Chinese-English Bilingual embeddings. - jina-embeddings-v2-base-de: 161 million parameters German-English Bilingual embeddings. - [jina-embeddings-v2-base-es](): Spanish-English Bilingual embeddings (soon). Data & Parameters Jina Embeddings V2 technical report Usage <details<summaryPlease apply mean pooling when integrating the model.</summary Why mean pooling? mean poooling takes all token embeddings from model output and averaging them at sentence/paragraph level. It has been proved to be the most effective way to produce high-quality sentence embeddings. We offer an encode function to deal with this. However, if you would like to do it without using the default encode function: You can use Jina Embedding models directly from transformers package. If you o
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | jinaai |
|---|---|
| Type | Embedding models |
| Parameters (lead) | 33M |
| Context | 8k tokens |
| Variants | 1 |
| Runs with | sentence-transformers |
| Released | 2023-09-27 |
| Popularity | 880k downloads / month |
| Likes | 142 |
| 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 |
|---|---|---|---|---|---|
| jina-embeddings-v2-small-en | 33M | BF16 | ~0.1 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 | 71.358 |
| Classification | MTEB AmazonCounterfactualClassification (en) | ap | 33.999 |
| Classification | MTEB AmazonCounterfactualClassification (en) | f1 | 65.385 |
| Classification | MTEB AmazonPolarityClassification | accuracy | 82.901 |
| Classification | MTEB AmazonPolarityClassification | ap | 78.014 |
| Classification | MTEB AmazonPolarityClassification | f1 | 82.834 |
| Classification | MTEB AmazonReviewsClassification (en) | accuracy | 40.89 |
| Classification | MTEB AmazonReviewsClassification (en) | f1 | 39.209 |
| Retrieval | MTEB ArguAna | map_at_1 | 23.257 |
| Retrieval | MTEB ArguAna | map_at_10 | 37.946 |
| Retrieval | MTEB ArguAna | map_at_100 | 39.17 |
| Retrieval | MTEB ArguAna | map_at_1000 | 39.181 |
| Retrieval | MTEB ArguAna | map_at_3 | 32.99 |
| Retrieval | MTEB ArguAna | map_at_5 | 35.468 |
| Retrieval | MTEB ArguAna | mrr_at_1 | 23.542 |
| Retrieval | MTEB ArguAna | mrr_at_10 | 38.057 |
| Retrieval | MTEB ArguAna | mrr_at_100 | 39.289 |
| Retrieval | MTEB ArguAna | mrr_at_1000 | 39.299 |
| Retrieval | MTEB ArguAna | mrr_at_3 | 33.096 |
| Retrieval | MTEB ArguAna | mrr_at_5 | 35.628 |
| Retrieval | MTEB ArguAna | ndcg_at_1 | 23.257 |
| Retrieval | MTEB ArguAna | ndcg_at_10 | 46.729 |
| Retrieval | MTEB ArguAna | ndcg_at_100 | 51.901 |
| Retrieval | MTEB ArguAna | ndcg_at_1000 | 52.16 |
Using it via the API
Once AxForge deploys jina-embeddings-small-en for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (jina-embeddings-small-en 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":"jina-embeddings-small-en","input":"text to embed"}'
Details
Languages
Trained / evaluated on
Tags
Papers
Licence
Open weights under apache-2.0 — commercial use is permitted. Deploy it on AxForge EU hardware on request. Read the licence ↗
Explore