🚀 Llama.cpp Quantizations of Nomic Embed Code: A State-of-the-Art Code Retriever

This project offers Llama.cpp quantizations of the Nomic Embed Code, a top - notch code retriever, enabling efficient code embedding and retrieval.

Blog | Technical Report | AWS SageMaker | Atlas Embedding and Unstructured Data Analytics Platform

Using llama.cpp commit 11683f579 for quantization.

Original model: nomic-embed-code

🚀 Quick Start

This model can be used with the llama.cpp server and other software that supports llama.cpp embedding models.

Queries embedded with nomic-embed-code must begin with the following prefix:

Represent this query for searching relevant code:

✨ Features

• High Performance: Outperforms Voyage Code 3 and OpenAI Embed 3 Large on CodeSearchNet.
• Multilingual Code Support: Trained for multiple programming languages (Python, Java, Ruby, PHP, JavaScript, Go).
• Advanced Architecture: 7B parameter code embedding model.
• Fully Open - Source: Model weights, training data, and evaluation code released.

📦 Installation

No specific installation steps are provided in the original document.

💻 Usage Examples

Basic Usage

Start a llama.cpp server:

llama-server -m nomic-embed-code.Q4_0.gguf --embeddings --pooling last

Advanced Usage

The following code shows how to use the prefix to embed user questions, e.g., in a RAG application.

import requests
from textwrap import dedent

def dot(va, vb):
    return sum(a*b for a, b in zip(va, vb))
def embed(texts):
    resp = requests.post('http://localhost:8080/v1/embeddings', json={'input': texts}).json()
    return [d['embedding'] for d in resp['data']]

docs = [
    dedent("""\
    def fn(n):
        if n < 0:
            raise ValueError
        return 1 if n == 0 else n * fn(n - 1)
    """).strip(),
    dedent("""\
    def fn(n):
        print(("Fizz" * (n % 3 == 0) + "Buzz" * (n % 5 == 0)) or n)
    """).strip(),
]
docs_embed = embed(docs)

query = 'Calculate the n-th factorial'
query_embed = embed(['Represent this query for searching relevant code: ' + query])[0]
print(f'query: {query!r}')
for d, e in zip(docs, docs_embed):
    print(f'\nsimilarity {dot(query_embed, e):.2f}:\n{d}')

You should see output similar to this:

query: 'Calculate the n-th factorial'

similarity 0.49:
def fn(n):
    if n < 0:
        raise ValueError
    return 1 if n == 0 else n * fn(n - 1)

similarity 0.32:
def fn(n):
    print(("Fizz" * (n % 3 == 0) + "Buzz" * (n % 5 == 0)) or n)

📚 Documentation

Download a file (not the whole branch)

Model Overview

nomic-embed-code is a state - of - the - art code embedding model that excels at code retrieval tasks.

Model Architecture

• Total Parameters: 7B
• Training Approach: Trained on the CoRNStack dataset with dual - consistency filtering and progressive hard negative mining.
• Supported Languages: Python, Java, Ruby, PHP, JavaScript, and Go.

CoRNStack Dataset Curation

Starting with the deduplicated Stackv2, we create text - code pairs from function docstrings and respective code. We filtered out low - quality pairs where the docstring wasn't English, too short, or that contained URLs, HTML tags, or invalid characters. We additionally kept docstrings with text lengths of 256 tokens or longer to help the model learn long - range dependencies.

After the initial filtering, we used dual - consistency filtering to remove potentially noisy examples. We embed each docstring and code pair and compute the similarity between each docstring and every code example. We remove pairs from the dataset if the corresponding code example is not found in the top - 2 most similar examples for a given docstring.

During training, we employ a novel curriculum - based hard negative mining strategy to ensure the model learns from challenging examples. We use a softmax - based sampling strategy to progressively sample hard negatives with increasing difficulty over time.

🔧 Technical Details

The model is trained on the CoRNStack dataset. The dataset curation process involves multiple steps of filtering to ensure high - quality text - code pairs. During training, dual - consistency filtering and progressive hard negative mining are used to improve the model's performance on code retrieval tasks.

📄 License

This project is licensed under the apache-2.0 license.

Join the Nomic Community

• Nomic Embed Ecosystem: https://www.nomic.ai/embed
• Website: https://nomic.ai
• Twitter: https://twitter.com/nomic_ai
• Discord: https://discord.gg/myY5YDR8z8

Citation

If you find the model, dataset, or training code useful, please cite our work:

@misc{suresh2025cornstackhighqualitycontrastivedata,
      title={CoRNStack: High-Quality Contrastive Data for Better Code Retrieval and Reranking},
      author={Tarun Suresh and Revanth Gangi Reddy and Yifei Xu and Zach Nussbaum and Andriy Mulyar and Brandon Duderstadt and Heng Ji},
      year={2025},
      eprint={2412.01007},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2412.01007},
}