subhadarship / kmeans_pytorch
kmeans using PyTorch
☆513Updated last year
Alternatives and similar repositories for kmeans_pytorch:
Users that are interested in kmeans_pytorch are comparing it to the libraries listed below
- This is a pytorch implementation of k-means clustering algorithm☆307Updated last month
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆223Updated last year
- PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566☆1,190Updated 2 years ago
- Pytorch implementation for t-SNE with cuda to accelerate☆332Updated 2 years ago
- Gaussian mixture models in PyTorch.☆555Updated last year
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆441Updated 2 years ago
- Learning Rate Warmup in PyTorch☆409Updated last month
- PyTorch implementation of the InfoNCE loss for self-supervised learning.☆541Updated last year
- An All-MLP solution for Vision, from Google AI☆1,019Updated 7 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆758Updated 11 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆426Updated 6 months ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.☆786Updated 11 months ago
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆583Updated 2 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,137Updated last year
- Approximating Wasserstein distances with PyTorch☆458Updated 2 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,070Updated 2 years ago
- A pytorch implementation for paper 'Exploring Simple Siamese Representation Learning'☆830Updated 2 years ago
- Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information☆329Updated 11 months ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057☆1,262Updated 3 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆494Updated last year
- A LARS implementation in PyTorch☆344Updated 5 years ago
- A pytorch implementation of MINE(Mutual Information Neural Estimation)☆345Updated 6 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆218Updated 3 years ago
- [arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis☆1,322Updated 4 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆491Updated 2 years ago
- Implementation of Linformer for Pytorch☆279Updated last year
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,056Updated 2 weeks ago
- An implementation of local windowed attention for language modeling☆443Updated 3 months ago
- ICLR 2021, Contrastive Learning with Hard Negative Samples☆265Updated 3 years ago