Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
☆234Dec 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:
- This is a pytorch implementation of k-means clustering algorithm☆345Mar 4, 2025Updated last year
- kmeans using PyTorch☆535May 9, 2023Updated 3 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
- Un-*** 50 billions multimodality dataset☆24Sep 14, 2022Updated 3 years ago
- ☆18Sep 27, 2021Updated 4 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Matlab implementation of "K-Nearest Neighbors Hashing" (CVPR2019)☆27Jun 15, 2019Updated 6 years ago
- Supplementary Code for Unsupervised Neural Quantization for Compressed-Domain Similarity Search☆36Aug 15, 2019Updated 6 years ago
- A torch-based implementation of K-Means and K-Means++☆17Dec 6, 2020Updated 5 years ago
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆174Feb 11, 2025Updated last year
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 3 years ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆442Mar 26, 2024Updated 2 years ago
- Fixed Radius Nearest Neighbor Search on GPU☆194Mar 6, 2024Updated 2 years ago
- Implementation of vector quantization algorithms, codes for Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner P…☆61Jan 7, 2021Updated 5 years ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Large dataset storage format for Pytorch☆45Aug 19, 2021Updated 4 years ago
- CUDA implementation of Hierarchical Navigable Small World Graph algorithm☆176Apr 19, 2021Updated 5 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,294Mar 3, 2024Updated 2 years ago
- Fast, differentiable sorting and ranking in PyTorch☆865Mar 3, 2026Updated 3 months ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆682Feb 21, 2020Updated 6 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Jul 16, 2020Updated 5 years ago
- Blazingly fast neighborhood attention☆14Nov 28, 2023Updated 2 years ago
- ☆45Oct 20, 2022Updated 3 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆70Mar 29, 2021Updated 5 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆13Nov 26, 2021Updated 4 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,130Apr 20, 2022Updated 4 years ago
- Implementation for paper "BATMANN: A Binarized-All-Through Memory-Augmented Neural Network for Efficient In-Memory Computing"☆12Jan 12, 2022Updated 4 years ago
- Topic Model based on Pretrained Sentence Embeddings (with BERT)☆13Feb 8, 2023Updated 3 years ago
- Implementation of Generative Moment Matching Networks in pytorch☆16Jan 12, 2019Updated 7 years ago
- PyTorch implementation of Refine and Represent: Region-to-Object Representation Learning.☆21Jun 19, 2025Updated 11 months ago
- The course of Parallel programming in Nankai university（南开大学《并行程序设计》课程 by 王 刚老师）☆12Oct 5, 2022Updated 3 years ago
- Deep Supervised Hashing with Triplet Labels☆10Nov 2, 2017Updated 8 years ago
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆81Dec 4, 2022Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Fast Point Cloud Segmentation Using Permutohedral Lattices☆16Mar 20, 2023Updated 3 years ago
- Hugging Face RoBERTa with Flash Attention 2☆24Sep 14, 2025Updated 9 months ago
- Pytorch library for fast transformer implementations☆1,771Mar 23, 2023Updated 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
- Training Transformer-XL on 128 GPUs☆141Jun 11, 2020Updated 6 years ago
- ☆21Apr 16, 2024Updated 2 years ago
- A lightweight command line interface for the management of arbitrary machine learning tasks☆19Jan 29, 2021Updated 5 years ago