DeMoriarty / fast_pytorch_kmeansLinks
This is a pytorch implementation of k-means clustering algorithm
☆315Updated 4 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☆516Updated 2 years ago
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆68Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆226Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆434Updated 10 months ago
- Implementation of Linformer for Pytorch☆294Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- An implementation of local windowed attention for language modeling☆466Updated 2 weeks ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆600Updated 8 months ago
- ☆136Updated last year
- Differentiable Top-k Classification Learning☆83Updated 2 years ago
- Gaussian mixture models in PyTorch.☆569Updated 2 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆357Updated 4 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 4 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆208Updated last year
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 3 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆78Updated 2 years ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆600Updated 9 months ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆196Updated 2 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆267Updated 3 years ago
- ☆464Updated 2 years ago
- Learning Rate Warmup in PyTorch☆410Updated last month
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pytorch☆350Updated last year
- A LARS implementation in PyTorch☆349Updated 5 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆54Updated 10 months ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆144Updated 2 years ago
- PyTorch implementation of the InfoNCE loss for self-supervised learning.☆570Updated last year
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆486Updated last year
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆589Updated 2 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago