Model reference · open weights

elephant-embeddings-text-small

Embeddings llm-semantic-router Embeddings 1 build Open weights 1k dl/mo

elephant-embeddings-text-small is an open-weight embedding model from llm-semantic-router. elephant-embeddings-v1-text-small (BF16) weighs 614 MB; the smallest configuration that runs it is RTX 3060 12 GB.

For self-hosting, compare this build's requirements below with the GPU you plan to rent.

  • elephant-embeddings-text-small is a multilingual embedding model with 307M parameters designed for sentence-similarity tasks in agent-native retrieval and memory workflows.
  • It supports a context length of 32,768 tokens and utilizes 2D Matryoshka training to allow dimension and layer reduction for varying latency and quality needs.
  • The model is distributed under the Apache 2.0 license.

Summary of the llm-semantic-router/elephant-embeddings-v1-text-small model card, 2026-10-01

What it is

Released byllm-semantic-router
Released2026-04-16
Parameters307M
VRAM614 MB for the weights

What it runs on

Memory and cards for elephant-embeddings-v1-text-small (BF16)

614 MBweights, file size
1.1 GBruntime overhead
CardRunsMemory
RTX 3060 12 GBfits11.6 GB
RTX 4060 Ti 16 GBfits15.4 GB
RTX 3090 24 GBfits23.4 GB
RTX 4090 24 GBfits23.4 GB
RTX 5090 32 GBfits31.0 GB
L40S 48 GBfits44.0 GB
A100 80 GBfits78.2 GB
H100 80 GBfits78.1 GB
RTX PRO 6000 Blackwell 96 GBfits93.8 GB
DGX Spark (GB10) 128 GB unifiedfits107 GB
H200 141 GBfits138 GB
B200 180 GBfits176 GB

From the model card

What llm-semantic-router says about elephant-embeddings-text-small

Read the model card

elephant-embeddings-v1-text-small is a multilingual long-context embedding model purpose-built for agent-native retrieval, memory, and decision workflows.

It is designed for systems where embeddings sit on the semantic hot path rather than at the edge of the stack: memory lookup, knowledge retrieval, tool matching, task routing, long-horizon recall, clustering, and multilingual indexing. Its value is not just a benchmark score, but a practical operating profile that fits real agent runtimes: 32K context, 2D Matryoshka adaptability across dimensions and layers, 307M-class deployability, and strong latency-quality efficiency under repeated inference.

In short, elephant-embeddings-v1-text-small is built for teams that want one embedding space to support fast routing, scalable retrieval, and high-confidence semantic matching without paying the operational cost of a much larger model.

Why it fits agentic workloads

Agentic systems do not call embeddings once. They call them everywhere: before retrieval, during routing, when matching tools, when searching memory, and while compressing or re-ranking state. That means a useful agent embedding model must be more than accurate — it must also be flexible under tight runtime budgets.

elephant-embeddings-v1-text-small is designed around that reality.

1. One model, many budget tiers

This model supports Matryoshka embeddings, which means you can encode once at full size and truncate to smaller dimensions with limited quality loss.

That is especially useful for agent systems because different stages of the stack often need different budgets:

  • 64d for very cheap candidate generation, broad routing, or huge memory banks
  • 256d for balanced retrieval over large corpora
  • 768d for highest-quality retrieval, offline indexing, or final-stage matching

Instead of managing separate embedding models for each tier, you can keep one semantic space and choose the dimensional budget that matches the task.

2. 2D Matryoshka gives runtime flexibility, not just storage savings

The model is trained with 2D Matryoshka behavior:

  • dimension reduction for smaller vectors and lower storage / bandwidth cost
  • layer reduction for lower-latency inference paths in custom runtimes

This matters for agents because the same system often mixes:

  • latency-sensitive routing decisions
  • high-volume memory scans
  • higher-quality retrieval for final evidence gathering

A single model that can serve multiple latency / quality profiles is much easier to operate than a stack of unrelated specialized encoders.

3. Long context helps when agent state is not naturally short

Many agent workloads are not short isolated queries. They involve:

  • tool descriptions
  • execution traces
  • long notes
  • merged memory summaries
  • multi-hop research snippets
  • large document chunks

With 32,768 tokens of context length, elephant-embeddings-v1-text-small can represent larger semantic units before you are forced into aggressive chunking. That helps preserve cross-section meaning in long documents and richer memory entries.

4. Small enough to be operationally practical

At roughly 307M parameters, this model sits in a useful middle ground:

  • substantially lighter than large embedding models in the 600M+ or multi-billion range
  • still expressive enough for multilingual retrieval and similarity work
  • easier to host in systems where embedding is part of a hot path rather than an occasional offline batch

For agentic platforms, that usually means better economics and simpler scaling.

5. One embedding space across the stack

Agent systems are easier to operate when routing, retrieval, memory search, and semantic matching all live in the same vector space.

elephant-embeddings-v1-text-small is well suited to that pattern:

  • 64d can serve broad routing and large-memory scanning
  • 256d can cover the main retrieval tier
  • 768d can stay reserved for the highest-fidelity matching paths

That means one model can cover multiple semantic stages without forcing the system to juggle incompatible encoders, duplicated indexes, or divergent retrieval behavior.

Model at a glance

FeatureValue
Parameters307M
ArchitectureModernBERT encoder with YaRN scaling
Hidden Size768
Layers22
Context Length32,768 tokens
PoolingMean pooling
SimilarityCosine
LanguagesMultilingual
Matryoshka Dimensions768, 512, 256, 128, 64

Headline results

MetricScore
MTEB Mean (24 tasks)61.4
STS Benchmark80.5
Dimension Retention99% @ 256d, 98% @ 64d
Layer Speedup3.3× @ 6L, 5.8× @ 3L
Latency vs BGE-M31.6-3.1× faster on longer sequences / larger batches

These numbers make the model particularly attractive for systems that must balance quality, latency, vector size, and deployment simplicity instead of optimizing only for leaderboard peak score.

Usage

Basic usage with Sentence Transformers

from sentence_transformers import SentenceTransformer

model = SentenceTransformer("/path/to/elephant-embeddings-v1-text-small")

texts = [
    "Find tool descriptions related to browser automation.",
    "检索和用户历史偏好相关的记忆。",
    "Retrieve notes about deployment failures in staging.",
]

embeddings = model.encode(texts)
print(embeddings.shape)  # (3, 768)

Matryoshka truncation for smaller vectors

import torch.nn.functional as F
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("/path/to/elephant-embeddings-v1-text-small")
embeddings = model.encode(texts, convert_to_tensor=True)

# Balanced retrieval tier
embeddings_256d = F.normalize(embeddings[:, :256], p=2, dim=1)

# Ultra-cheap routing / large memory-bank tier
embeddings_6

Quoted from the model card on Hugging Face. The full card is behind the Hugging Face link above.

Benchmarks

Reported results

As published on the model card: the maker's own numbers, not measured by AxForge.

TaskDatasetMetricScore
STSSTS Benchmarkspearman80.500
© 2026 AxForge · EU-hosted AI infrastructure Pricing Docs Trust Privacy Terms