Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
☆236Dec 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pure python implementation of product quantization for nearest neighbor search☆362Jun 7, 2025Updated last year
- ☆18Sep 27, 2021Updated 4 years ago
- Supplementary Code for Unsupervised Neural Quantization for Compressed-Domain Similarity Search☆36Aug 15, 2019Updated 7 years ago
- A torch-based implementation of K-Means and K-Means++☆17Dec 6, 2020Updated 5 years ago
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆445Mar 26, 2024Updated 2 years ago
- Large dataset storage format for Pytorch☆44Aug 19, 2021Updated 5 years ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆175Feb 11, 2025Updated last year
- Implementation of vector quantization algorithms, codes for Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner P…☆61Jan 7, 2021Updated 5 years ago
- Fixed Radius Nearest Neighbor Search on GPU☆194Mar 6, 2024Updated 2 years ago
- CUDA implementation of Hierarchical Navigable Small World Graph algorithm☆179Apr 19, 2021Updated 5 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,293Mar 3, 2024Updated 2 years ago
- Fast, differentiable sorting and ranking in PyTorch☆866Mar 3, 2026Updated 5 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- A simple semantic search engine for scientific papers.☆28Sep 14, 2023Updated 2 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆682Feb 21, 2020Updated 6 years ago
- Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning☆13Feb 23, 2021Updated 5 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Jul 16, 2020Updated 6 years ago
- Continual Learning with Gated Incremental Memories for Sequential Data Processing. IJCNN 2020. Continual Learning with Recurrent Neural N…☆15Oct 13, 2021Updated 4 years ago
- ☆45Oct 20, 2022Updated 3 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆70Mar 29, 2021Updated 5 years ago
- ☆13Nov 26, 2021Updated 4 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,133Apr 20, 2022Updated 4 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- PyTorch implementation of Refine and Represent: Region-to-Object Representation Learning.☆21Jun 19, 2025Updated last year
- Topic Model based on Pretrained Sentence Embeddings (with BERT)☆13Feb 8, 2023Updated 3 years ago
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆82Dec 4, 2022Updated 3 years ago
- The course of Parallel programming in Nankai university（南开大学《并行程序设计》课程 by 王刚老师）☆12Oct 5, 2022Updated 3 years ago
- ☆17May 8, 2024Updated 2 years ago
- Hugging Face RoBERTa with Flash Attention 2☆23Sep 14, 2025Updated 11 months ago
- Pytorch library for fast transformer implementations☆1,777Mar 23, 2023Updated 3 years ago
- ❤️ CUDA/C++ GPU graph analytics simplified.☆32Sep 19, 2022Updated 3 years ago
- Unofficial PyTorch implementation of "Filter Response Normalization Layer: Eliminating Batch Dependence in the Training of Deep Neural Ne…☆22Dec 19, 2019Updated 6 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Training Transformer-XL on 128 GPUs☆140Jun 11, 2020Updated 6 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,189Jul 17, 2026Updated last month
- MODS with external deep descriptors/detectors☆54Oct 30, 2023Updated 2 years ago
- Source code for paper "Spectral Hashing" on NIPS-2009☆13Jan 7, 2020Updated 6 years ago
- Fast and memory-efficient clustering☆267Oct 16, 2023Updated 2 years ago
- Jittor code for Line Drawings for Face Portraits from Photos using Global and Local Structure based GANs (TPAMI)☆14Apr 19, 2021Updated 5 years ago
- Implementation and Benchmark Splits to study Out-of-Distribution Generalization in Deep Metric Learning.☆25Oct 2, 2021Updated 4 years ago