DeMoriarty / fast_pytorch_kmeansLinks
This is a pytorch implementation of k-means clustering algorithm
☆328Updated 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☆520Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆228Updated last year
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆73Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆438Updated last year
- Implementation of Linformer for Pytorch☆299Updated last year
- ☆142Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆617Updated 10 months ago
- An implementation of local windowed attention for language modeling☆483Updated 3 months ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆268Updated 4 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆492Updated last year
- Differentiable Top-k Classification Learning☆86Updated 2 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆370Updated 4 years ago
- Learning Rate Warmup in PyTorch☆413Updated 4 months ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆608Updated last year
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆801Updated last year
- Implementation of Slot Attention from GoogleAI☆464Updated last year
- Gaussian mixture models in PyTorch.☆582Updated 2 years ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆457Updated 3 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 4 years ago
- An implementation of the efficient attention module.☆321Updated 4 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆228Updated 3 years ago
- Compare neural networks by their feature similarity☆374Updated 2 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆196Updated 2 years ago
- A LARS implementation in PyTorch☆352Updated 5 years ago
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆257Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆140Updated 3 years ago
- ☆164Updated 2 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆55Updated last year