DeMoriarty / fast_pytorch_kmeans
This is a pytorch implementation of k-means clustering algorithm
☆297Updated 2 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
- kmeans using PyTorch☆502Updated last year
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆216Updated last year
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆62Updated 2 years ago
- Implementation of Linformer for Pytorch☆266Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆420Updated 4 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆557Updated 2 months ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆317Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆124Updated 11 months ago
- Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information☆323Updated 9 months ago
- Implementation of Slot Attention from GoogleAI☆409Updated 5 months ago
- An implementation of local windowed attention for language modeling☆412Updated last month
- ☆163Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- PyTorch implementation of the InfoNCE loss for self-supervised learning.☆516Updated last year
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆578Updated 2 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆217Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆738Updated 9 months ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆561Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆452Updated last year
- Pytorch implementation of set transformer☆562Updated 5 years ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆437Updated 2 years ago
- Official Open Source code for "Masked Autoencoders As Spatiotemporal Learners"☆327Updated 2 months ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆257Updated 3 years ago
- Reproduce CKA: Similarity of Neural Network Representations Revisited☆297Updated 4 years ago
- Differentiable Top-k Classification Learning☆79Updated 2 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆198Updated 9 months ago
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆337Updated 2 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆697Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago