KevinMusgrave / pytorch-metric-learning
Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.
Why this job: works in the areas we cover, but does not say which job it does
What it does
Train embedding models with composable PyTorch losses, hard-example miners, distances, reducers, and regularizers. Use them in your own training loop for retrieval, clustering, recognition, or learning from pairs and triplets.
Get started
uv pip install pytorch-metric-learning
pip install pytorch-metric-learning- Maintenance
- Active
- Last commit 4 days ago
- Stars
- 6,343
- +1 this week
- Licence
- MIT
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- v3.0.2
- Released 4 days ago
- Built with
- Python
- 99% of the code
Is it looked after?
- Contributors
- 2
- Last 90 days
- Issues
- 100% closed
- 0 still open
- Pull requests
- 153 merged
- 0 waiting
- Releases
- 62
- 8 in the last year
- Contributing guide (opens in new tab)
- Security policy (opens in new tab)
- No code of conduct
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| KaiyangZhou / deep-person-reid | 4,944 | 10 months ago | MIT | No downloads |
| scikit-learn-contrib / metric-learn | 1,430 | 7 months ago | MIT | No downloads |
| qdrant / quaterion | 666 | 2 months ago | Apache-2.0 | No downloads |
More: languages, topics, project record and preview image
Languages
- Python 99%
- PowerShell 1%
- Shell <1%
- HTML <1%
Topics
- metric-learning
- deep-learning
- computer-vision
- machine-learning
- pytorch
- deep-metric-learning
- image-retrieval
- self-supervised-learning
- contrastive-learning
- embeddings
Project record
- JobOther AI tools
- StatusListed
- Commits1,534
- Forks663
- Watching59
- Stars in 28 days+8
- CreatedOct 23, 2019
- First seen by usNov 7, 2026
- CheckedNov 9, 2026