SAVRN
Search Contact SAVRN

Open-weight model · Feature extraction

UAE-Large-V1

by WhereIsAI WhereIsAI/UAE-Large-V1

WhereIsAI/UAE-Large-V1 is licensed under MIT. Feel free to use it in any scenario. If you use it for academic papers, you could cite us via citation info. Welcome to using AnglE to train and infer powerful sentence embeddings.

Parameters335M
Context512
Weights5.4 GB
Licensemit
AccessOpen weights
Monthly Downloads1.1M

Runs On

What it takes to serve UAE-Large-V1 (335M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.7 GB 0.8 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.3 GB 0.4 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.2 GB 0.2 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Sep 18, 2026.

SAVRN's Notes on UAE-Large-V1

We would put this one to work turning English sentences into vectors for search and retrieval, which is what a 335M-parameter BERT embedding model with a 512-token window is built for. WhereIsAI trained it with the AnglE method in arXiv 2309.12871 and reports an MTEB average of 64.64. Memory is not a decision point: 0.8 GB at 16-bit, 0.4 GB at 8-bit, 0.2 GB at 4-bit. On the cheapest setup, one 192 GB MI300X at $1.85 an hour, plan around batches, not fit.

The MIT license permits commercial use, modification and redistribution with the copyright and permission notices intact, and access is open. Check the 512-token cap against your chunking, since longer inputs are cut. The 5.36 GB download across 20 files covers safetensors, onnx and openvino, so pull only the format your stack needs. A sibling, UAE-Code-Large-V1, covers code similarity. No host prices are on the Index yet.

Model Card

By WhereIsAI, published under mit, revision 9c9b2c999b33.

Universal AnglE Embedding

WhereIsAI/UAE-Large-V1 is licensed under MIT. Feel free to use it in any scenario. If you use it for academic papers, you could cite us viacitation info.

Follow us on:

Welcome to using AnglE to train and infer powerful sentence embeddings.

Achievements

  • May 16, 2024 | AnglE's paper is accepted by ACL 2024 Main Conference
  • Dec 4, 2023 | Our universal English sentence embeddingWhereIsAI/UAE-Large-V1 achieves SOTA on the MTEB Leaderboard with an average score of 64.64!

Siblings:

Usage

1. angle_emb

python -m pip install -U angle-emb

1) Non-Retrieval Tasks

There is no need to specify any prompts.

Read the full model card (345 words)

Configuration

Architecture
BertModel
Context length (tokens)
512
Layers
24
Hidden size
1,024
Feed-forward size
4,096
Attention heads
16
Vocabulary size
30,522
Stored precision
float32
Model type
bert

Identity and Version

Repository
WhereIsAI/UAE-Large-V1
Publisher
WhereIsAI
Task
Feature extraction
Modality
Text
Library
sentence-transformers
Parameters
335M parameters
Languages
en
Revision
9c9b2c999b3350cfb3171ed429320668e39b00b8
First published
2023-12-04
Last updated
2025-07-29

Files and Weights

20 files, 5.4 GB in total. The weights are 6 files totalling 5.4 GB in bin, onnx, safetensors.

Weights6 files · 5.4 GB
Configuration7 files · 253.9 KB
Tokenizer3 files · 944.1 KB
Documentation1 file · 66.2 KB
Other2 files · 2.0 MB
Repository1 file · 1.5 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights1.3 GB 8ac0e0e2eb9f
onnx/model.onnxWeights1.3 GB 8f00f2ff2028
onnx/model_fp16.onnxWeights668.8 MB 2841fe07ad38
onnx/model_quantized.onnxWeights337.0 MB b2c10399acd8
openvino/openvino_model.binWeights1.3 GB 2947cef8b69c
openvino/openvino_model_qint8_quantized.binWeights336.8 MB 64ee53b3bac2
1_Pooling/config.jsonConfiguration297 B
WhereIsAI_UAE-Large-V1.jsonConfiguration252.3 KB
config.jsonConfiguration655 B
config_sentence_transformers.jsonConfiguration171 B
modules.jsonConfiguration229 B
sentence_bert_config.jsonConfiguration53 B
special_tokens_map.jsonConfiguration125 B
README.mdDocumentation66.2 KB
openvino/openvino_model.xmlOther708.1 KB
openvino/openvino_model_qint8_quantized.xmlOther1.3 MB
.gitattributesRepository1.5 KB
tokenizer.jsonTokenizer711.4 KB
tokenizer_config.jsonTokenizer1.2 KB
vocab.txtTokenizer231.5 KB

License and Download

License
mit
Access
Open weights, no gate
Download size
5.4 GB
Download from WhereIsAI

Released by WhereIsAI through its official repository on Hugging Face. Read the license.

Built From

Evaluations

Each result is shown as reported, with the conditions its reporter stated. None is a SAVRN measurement. A comparison lines two results up only when their configuration, unit and setup are all stated and identical.

BenchmarkConditionsResultReported byRevisionDate
MTEB AmazonCounterfactualClassification (en) Configuration enTask ClassificationMetric accuracyComparison conditions not established 75.5522 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonCounterfactualClassification (en) Configuration enTask ClassificationMetric apComparison conditions not established 38.2641 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonCounterfactualClassification (en) Configuration enTask ClassificationMetric f1Comparison conditions not established 69.4098 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonPolarityClassification Configuration defaultTask ClassificationMetric accuracyComparison conditions not established 92.8427 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonPolarityClassification Configuration defaultTask ClassificationMetric apComparison conditions not established 89.5757 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonPolarityClassification Configuration defaultTask ClassificationMetric f1Comparison conditions not established 92.8259 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonReviewsClassification (en) Configuration enTask ClassificationMetric accuracyComparison conditions not established 48.292 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB AmazonReviewsClassification (en) Configuration enTask ClassificationMetric f1Comparison conditions not established 47.9026 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_1Comparison conditions not established 42.105 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_10Comparison conditions not established 58.181 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_100Comparison conditions not established 58.654 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_1000Comparison conditions not established 58.657 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_3Comparison conditions not established 54.386 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric map_at_5Comparison conditions not established 56.758 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_1Comparison conditions not established 42.745 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_10Comparison conditions not established 58.437 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_100Comparison conditions not established 58.895 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_1000Comparison conditions not established 58.898 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_3Comparison conditions not established 54.635 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric mrr_at_5Comparison conditions not established 57 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_1Comparison conditions not established 42.105 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_10Comparison conditions not established 66.15 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_100Comparison conditions not established 68.048 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_1000Comparison conditions not established 68.114 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_3Comparison conditions not established 58.477 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric ndcg_at_5Comparison conditions not established 62.768 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_1Comparison conditions not established 42.105 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_10Comparison conditions not established 9.111 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_100Comparison conditions not established 0.991 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_1000Comparison conditions not established 0.1 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_3Comparison conditions not established 23.447 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric precision_at_5Comparison conditions not established 16.159 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_1Comparison conditions not established 42.105 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_10Comparison conditions not established 91.11 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_100Comparison conditions not established 99.147 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_1000Comparison conditions not established 99.644 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_3Comparison conditions not established 70.341 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArguAna Configuration defaultTask RetrievalMetric recall_at_5Comparison conditions not established 80.797 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArxivClusteringP2P Configuration defaultTask ClusteringMetric v_measureComparison conditions not established 49.0258 WhereIsAI
Publisher reported
Evaluated revision not stated
MTEB ArxivClusteringS2S Configuration defaultTask ClusteringMetric v_measureComparison conditions not established 43.0936 WhereIsAI
Publisher reported
Evaluated revision not stated

Memory Requirements

PrecisionWeights in memory
As published5.4 GB
16-bit0.7 GB
8-bit0.3 GB
4-bit0.2 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Compare UAE-Large-V1

Questions About UAE-Large-V1

How much GPU memory does UAE-Large-V1 need?

About 0.8 GB at 16-bit and 0.2 GB at 4-bit: the weights (335M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run UAE-Large-V1 on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use UAE-Large-V1 commercially?

Yes. UAE-Large-V1 is released under MIT License. The MIT License is a short permissive license. It permits commercial use, modification and redistribution, provided the copyright notice and permission notice are included.

What is UAE-Large-V1's context length?

512 tokens, from the maximum position embeddings in its published configuration.

Similar Models

Model · Feature extraction

mxbai-embed-large-v1

Mixedbread

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. 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. If you haven't already, you can install the Transformers.js JavaScript library from NPM using: You can then…

Open weights apache-2.0 335M parameters 512 tokens sentence-transformers

For more details please refer to our Github: FlagEmbedding. If you are looking for a model that supports more languages, longer texts, and other retrieval methods, you can try using bge-m3. FlagEmbedding focuses on retrieval-augmented LLMs, consisting of the following projects currently: - 1/30/2024: Release BGE-M3, a new member to BGE model series! M3 stands for Multi-linguality (100+ languages), Multi-granularities (input length up to 8192), Multi-Functionality (unification of dense, lexical, multi-vec/colbert retrieval). It is the first embedding model that supports all three retrieval methods, achieving new SOTA on multi-lingual (MIRACL) and cross-lingual (MKQA) benchmarks. Technical…

Open weights mit 335M parameters 512 tokens sentence-transformers

Model · Feature extraction

jina-embeddings-v2-base-code

Jina AI

The easiest way to starting using jina-embeddings-v2-base-code is to use Jina AI's Embedding API. jina-embeddings-v2-base-code is an multilingual embedding model speaks English and 30 widely used programming languages. Same as other jina-embeddings-v2 series, it supports 8192 sequence length. jina-embeddings-v2-base-code 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-base-code is pretrained on the github-code dataset. The model is further trained on Jina AI's collection of more than 150 millions of coding question answer and docstring source code pairs. These pairs were obtained…

Open weights apache-2.0 161M parameters 8,192 tokens sentence-transformers

Model · Feature extraction

Vela-1.0-Omni-Nano

vLLM Semantic Router

Vela Omni Nano maps text, images, and speech into a shared embedding space for multimodal search, routing, and use a 0–100 scale; higher is better. All applicable models use the same examples and retrieval pools. N/A denotes a modality the text-only model does not support. Bold Vela scores improve on multi-modal-embed-small. Macro-F1 gives equal weight to every intent class (77 for Banking77 and 60 for MASSIVE), complementing the query-weighted accuracy; undefined class F1 is zero. Text evaluation uses fixed class prototypes: 3,080 Banking77 and 2,972 MASSIVE English queries. Vela Omni is adapted using training examples and intent labels from these two datasets; comparison models are…

Open weights apache-2.0 134M parameters pytorch

Model · Feature extraction

multilingual-e5-large-instruct

Liang Wang

Liang Wang, Nan Yang, Xiaolong Huang, Linjun Yang, Rangan Majumder, Furu Wei, arXiv 2024 This model has 24 layers and the embedding size is 1024. Below are examples to encode queries and passages from the MS-MARCO passage ranking dataset. Usage with Infinity: This model is initialized from xlm-roberta-large and continually trained on a mixture of multilingual datasets. It supports 100 languages from xlm-roberta, but low-resource languages may see performance degradation. First stage: contrastive pre-training with 1 billion weakly supervised text pairs. Check out unilm/e5 to reproduce evaluation results on the BEIR and MTEB benchmark. 1. Do I need to add instructions to the query? Yes, this…

Open weights mit 560M parameters 514 tokens sentence-transformers

Model · Feature extraction

multilingual-e5-large

Liang Wang

Liang Wang, Nan Yang, Xiaolong Huang, Linjun Yang, Rangan Majumder, Furu Wei, arXiv 2024 This model has 24 layers and the embedding size is 1024. Below is an example to encode queries and passages from the MS-MARCO passage ranking dataset. This model is initialized from xlm-roberta-large and continually trained on a mixture of multilingual datasets. It supports 100 languages from xlm-roberta, but low-resource languages may see performance degradation. For all labeled datasets, we only use its training set for fine-tuning. For other training details, please refer to our paper at https://arxiv.org/pdf/2402.05672. Check out unilm/e5 to reproduce evaluation results on the BEIR and MTEB…

Open weights mit 560M parameters 514 tokens sentence-transformers