tensorflow/similarity

Health v2 · maintenance only

Flags: archived

Adoption not part of the score

1020 stars · 105 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

TensorFlow Similarity is a Python library built on TensorFlow for training and serving similarity and metric learning models, including self-supervised and contrastive methods. It provides losses, metrics, samplers, visualizers, and an indexing subsystem for embedding-based nearest neighbor search.

Use cases

• train a metric learning model to output embeddings for image similarity search
• pretrain a model with self-supervised contrastive learning on unlabeled images
• find nearest neighbors of images in a large corpus using learned embeddings
• cluster unseen examples by similarity after training on few classes
• fine-tune a contrastively pretrained model on a small labeled dataset
• evaluate retrieval quality with metrics like Precision@K

When to choose

• you use TensorFlow and need metric or contrastive learning with losses like SimCLR, Barlow Twins, or VicReg
• you want an integrated pipeline from training embeddings to indexing and querying nearest neighbors
• you need self-supervised pretraining to boost accuracy on small labeled datasets

When to avoid

• you work in PyTorch or another framework rather than TensorFlow
• you need a stable, production-guaranteed API - the library is in beta and may have breaking changes
• you only need off-the-shelf embeddings without training your own similarity model

Facets

library · maturity maintenance

machine-learning deep-learning search-engine data-visualization machine-learning deep-learning computer-vision python similarity-learning metric-learning contrastive-learning self-supervised-learning embeddings nearest-neighbor-search tensorflow knn simclr barlow-twins search

Member repositories

For agents

markdown · JSON · MCP: product_card(name="tensorflow/similarity")