ncorder/llama-embed-nemotron-8b-mlx-fp16

ncorder/llama-embed-nemotron-8b-mlx-fp16

MLX conversion of `nvidia/llama-embed-nemotron-8b` — the #1 embedding model under 20B parameters on the MTEB leaderboard, outperforming models 3x its size.

• —Parameters: 7.5B
• —Quantization: None (float16)
• —Model size: 15 GB
• —Architecture: Llama-3.1-8B with bidirectional attention
• —Embedding dimension: 4096
• —Max sequence length: 32,768 tokens
• —Converted with: mlx-embeddings

Full float16 precision. Near-identical to the original bfloat16 weights.

All variants

Reference (original bf16 PyTorch): relevant=0.3771, irrelevant=0.0581, margin=0.3190

Usage

pip install mlx-embeddings

from mlx_embeddings.utils import load
import mlx.core as mx

model, tokenizer = load("ncorder/llama-embed-nemotron-8b-mlx-fp16")

query = "Instruct: Given a question, retrieve passages that answer the question\nQuery: How do neural networks learn patterns from examples?"
document = "Deep learning models adjust their weights through backpropagation."

def embed(text):
    inputs = tokenizer(text, return_tensors="np", padding=True)
    out = model(
        mx.array(inputs["input_ids"]),
        mx.array(inputs["attention_mask"])
    )
    return out.text_embeds

q_emb = embed(query)
d_emb = embed(document)
score = (q_emb @ d_emb.T).item()
print(f"Similarity: {score:.4f}")

Query formatting

This model is instruction-aware. For retrieval, prefix queries with:

Instruct: {task_instruction}
Query: {your_query}

Documents are embedded without any prefix.

License

This model inherits the NVIDIA license from the original — research/non-commercial use only. Also subject to the Llama 3.1 Community License.

Credits

• —Original model by NVIDIA: `nvidia/llama-embed-nemotron-8b`
• —Conversion via `mlx-embeddings` by Prince Canuma
• —Technical report