DeMoriarty / fast_pytorch_kmeansLinks
This is a pytorch implementation of k-means clustering algorithm
☆325Updated 7 months ago
Alternatives and similar repositories for fast_pytorch_kmeans
Users that are interested in fast_pytorch_kmeans are comparing it to the libraries listed below
Sorting:
- kmeans using PyTorch☆519Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆228Updated last year
- Implementation of Linformer for Pytorch☆300Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆437Updated last year
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆72Updated 2 years ago
- An implementation of local windowed attention for language modeling☆480Updated 2 months ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆141Updated last year
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆367Updated 4 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆798Updated last year
- Gaussian mixture models in PyTorch.☆577Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆490Updated last year
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆227Updated 3 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆195Updated 2 years ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆605Updated 11 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆611Updated 10 months ago
- A simple cross attention that updates both the source and target in one step☆183Updated 2 months ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆269Updated 4 years ago
- Implementation of Slot Attention from GoogleAI☆461Updated last year
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 4 years ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆454Updated 3 years ago
- Differentiable Top-k Classification Learning☆86Updated 2 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- A LARS implementation in PyTorch☆352Updated 5 years ago
- Learning Rate Warmup in PyTorch☆412Updated 3 months ago
- ☆164Updated 2 years ago
- An implementation of the efficient attention module.☆321Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 3 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆208Updated last year
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆370Updated 2 years ago