DeMoriarty / TorchPQ
Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
☆219Updated last year
Alternatives and similar repositories for TorchPQ:
Users that are interested in TorchPQ are comparing it to the libraries listed below
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆259Updated 3 years ago
- ☆164Updated 2 years ago
- Fully featured implementation of Routing Transformer☆289Updated 3 years ago
- A LARS implementation in PyTorch☆342Updated 5 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 2 years ago
- Block-sparse primitives for PyTorch☆153Updated 3 years ago
- This is a pytorch implementation of k-means clustering algorithm☆301Updated last week
- Sequence modeling with Mega.☆296Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆372Updated 4 years ago
- Demystify RAM Usage in Multi-Process Data Loaders☆189Updated last year
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- ☆375Updated last year
- Differentiable Sorting Networks☆112Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated 2 years ago
- Implementation of Linformer for Pytorch☆272Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆126Updated 2 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆747Updated last year
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆378Updated 11 months ago
- Understanding Training Dynamics of Deep ReLU Networks☆284Updated last month
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Fast, differentiable sorting and ranking in PyTorch☆794Updated last year
- Experimental ground for optimizing memory of pytorch models☆365Updated 6 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆746Updated 10 months ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆454Updated last year
- Differentiable Top-k Classification Learning☆80Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago