Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
☆232Dec 12, 2023Updated 2 years ago
Alternatives and similar repositories for TorchPQ
Users that are interested in TorchPQ are comparing it to the libraries listed below
Sorting:
- This is a pytorch implementation of k-means clustering algorithm☆341Mar 4, 2025Updated last year
- kmeans using PyTorch☆531May 9, 2023Updated 2 years ago
- ☆18Sep 27, 2021Updated 4 years ago
- Code for papers "Hashing with Mutual Information" (TPAMI 2019) and "Hashing with Binary Matrix Pursuit" (ECCV 2018)☆19Oct 11, 2019Updated 6 years ago
- Large dataset storage format for Pytorch☆45Aug 19, 2021Updated 4 years ago
- Jittor code for Line Drawings for Face Portraits from Photos using Global and Local Structure based GANs (TPAMI)☆14Apr 19, 2021Updated 4 years ago
- A simple semantic search engine for scientific papers.☆28Sep 14, 2023Updated 2 years ago
- Supplementary Code for Unsupervised Neural Quantization for Compressed-Domain Similarity Search☆35Aug 15, 2019Updated 6 years ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆427Mar 26, 2024Updated last year
- ☆28Jun 9, 2025Updated 8 months ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 3 years ago
- Lightweight knowledge distillation pipeline☆28Nov 29, 2021Updated 4 years ago
- Matlab implementation of "K-Nearest Neighbors Hashing" (CVPR2019)☆27Jun 15, 2019Updated 6 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Jul 16, 2020Updated 5 years ago
- Official PyTorch implementation of Time-aware Large Kernel (TaLK) Convolutions (ICML 2020)☆29Dec 9, 2020Updated 5 years ago
- ☆21Apr 16, 2024Updated last year
- ☆17May 8, 2024Updated last year
- Fast, differentiable sorting and ranking in PyTorch☆852Updated this week
- Fixed Radius Nearest Neighbor Search on GPU☆192Mar 6, 2024Updated 2 years ago
- An open source implementation of CLIP.☆33Nov 7, 2022Updated 3 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,295Mar 3, 2024Updated 2 years ago
- Pure python implementation of product quantization for nearest neighbor search☆360Jun 7, 2025Updated 9 months ago
- MODS with external deep descriptors/detectors☆54Oct 30, 2023Updated 2 years ago
- ☆23Jun 25, 2021Updated 4 years ago
- Un-*** 50 billions multimodality dataset☆23Sep 14, 2022Updated 3 years ago
- Hugging Face RoBERTa with Flash Attention 2☆24Sep 14, 2025Updated 5 months ago
- Rank-DistiLLM: Closing the Effectiveness Gap Between Cross-Encoders and LLMs for Passage Re-Ranking☆25Apr 4, 2025Updated 11 months ago
- ☆117Nov 21, 2023Updated 2 years ago
- Pytorch library for fast transformer implementations☆1,763Mar 23, 2023Updated 2 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated last month
- Using large language models to maintain AI_CHANGELOG.md☆14Jul 15, 2024Updated last year
- TSDG: An efficient index graph for graph-based nearest neighbor search☆10Jul 14, 2022Updated 3 years ago
- Code for the paper "Faster Neural Network Training with Approximate Tensor Operations"☆10Oct 23, 2021Updated 4 years ago
- This repository provides the code for replicating the experiments in the paper "Building One-Shot Semi-supervised (BOSS) Learning up to F…☆36Aug 9, 2020Updated 5 years ago
- Differentiable Data Augmentation Library☆125Aug 7, 2022Updated 3 years ago
- DISTWAR-enabled rasterization engine for the paper "3D Gaussian Splatting for Real-Time Rendering of Radiance Fields"☆22Jan 16, 2024Updated 2 years ago
- PyTorch code for the EMNLP 2020 paper "Embedding Words in Non-Vector Space with Unsupervised Graph Learning"☆41Feb 18, 2021Updated 5 years ago
- [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization☆93May 5, 2022Updated 3 years ago
- Manually implemented quantization-aware training☆23Oct 12, 2022Updated 3 years ago