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. 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☆343Mar 4, 2025Updated last year
- kmeans using PyTorch☆534May 9, 2023Updated 2 years ago
- Un-*** 50 billions multimodality dataset☆24Sep 14, 2022Updated 3 years ago
- A torch-based implementation of K-Means and K-Means++☆17Dec 6, 2020Updated 5 years ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆437Mar 26, 2024Updated 2 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Fixed Radius Nearest Neighbor Search on GPU☆192Mar 6, 2024Updated 2 years ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- Large dataset storage format for Pytorch☆45Aug 19, 2021Updated 4 years ago
- CUDA implementation of Hierarchical Navigable Small World Graph algorithm☆175Apr 19, 2021Updated 5 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,296Mar 3, 2024Updated 2 years ago
- Customized matrix multiplication kernels☆57Mar 5, 2022Updated 4 years ago
- Fast, differentiable sorting and ranking in PyTorch☆861Mar 3, 2026Updated 2 months ago
- A simple semantic search engine for scientific papers.☆28Sep 14, 2023Updated 2 years ago
- VB Diarization with Eigenvoice and HMM Priors, refactored☆15Jul 27, 2021Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆682Feb 21, 2020Updated 6 years ago
- ☆30Jun 9, 2025Updated 10 months 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
- ICME2022 Special Session “Beyond Accuracy: Responsible, Responsive, and Robust Multimedia Retrieval ”☆12Jun 3, 2024Updated last year
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,128Apr 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
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Topic Model based on Pretrained Sentence Embeddings (with BERT)☆13Feb 8, 2023Updated 3 years ago
- A set of scripts to use in preparing a corpus for speech-to-text processing with the Kaldi Automatic Speech Recognition Library.☆15May 19, 2020Updated 5 years ago
- PyTorch implementation of Refine and Represent: Region-to-Object Representation Learning.☆21Jun 19, 2025Updated 10 months ago
- Experiment of using Tangent to autodiff triton☆82Jan 22, 2024Updated 2 years ago
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆76Dec 4, 2022Updated 3 years ago
- Fast Point Cloud Segmentation Using Permutohedral Lattices☆16Mar 20, 2023Updated 3 years ago
- Hugging Face RoBERTa with Flash Attention 2☆24Sep 14, 2025Updated 7 months ago
- PyTorch implementation of Metric-Guided Prototype Learning for hierarchical classification.☆29Oct 20, 2021Updated 4 years ago
- Pytorch library for fast transformer implementations☆1,769Mar 23, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- 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 5 years ago
- Generate an FPGA design for a TWN☆11Nov 4, 2019Updated 6 years ago
- Fast and memory-efficient clustering☆264Oct 16, 2023Updated 2 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,170Updated this week
- 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