Alternatives and similar repositories for kmeans_pytorch

Users that are interested in kmeans_pytorch are comparing it to the libraries listed below

• DeMoriarty / fast_pytorch_kmeans
  This is a pytorch implementation of k-means clustering algorithm
  ☆339Updated 11 months ago
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆444Updated last year
• mxl1990 / tsne-pytorch
  Pytorch implementation for t-SNE with cuda to accelerate
  ☆340Updated 2 years ago
• ldeecke / gmm-torch
  Gaussian mixture models in PyTorch.
  ☆584Updated 2 years ago
• WeiChengTseng / Pytorch-PCGrad
  Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
  ☆394Updated 4 years ago
• RElbers / info-nce-pytorch
  PyTorch implementation of the InfoNCE loss for self-supervised learning.
  ☆607Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,056Updated 6 months ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆415Updated 7 months ago
• facebookresearch / simsiam
  PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566
  ☆1,225Updated 3 years ago
• salesforce / PCL
  PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
  ☆610Updated 9 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆825Updated last year
• ssnl / align_uniform
  Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…
  ☆460Updated 3 years ago
• Linear95 / CLUB
  Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information
  ☆353Updated last year
• jefflai108 / Contrastive-Predictive-Coding-PyTorch
  Contrastive Predictive Coding for Automatic Speaker Verification
  ☆506Updated 6 years ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆469Updated 2 years ago
• jokofa / torch_kmeans
  PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
  ☆80Updated 3 years ago
• DeMoriarty / TorchPQ
  Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
  ☆231Updated 2 years ago
• Spijkervet / SimCLR
  PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
  ☆816Updated last year
• rishikksh20 / FNet-pytorch
  Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms
  ☆261Updated 4 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆306Updated 2 years ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆616Updated last year
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆216Updated 4 years ago
• sthalles / PyTorch-BYOL
  PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
  ☆502Updated 3 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆824Updated 3 years ago
• cmsflash / efficient-attention
  An implementation of the efficient attention module.
  ☆328Updated 5 years ago
• AndrewAtanov / simclr-pytorch
  PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results
  ☆211Updated last year
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆261Updated 5 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆352Updated 5 years ago
• joshr17 / HCL
  ICLR 2021, Contrastive Learning with Hard Negative Samples
  ☆279Updated 4 years ago
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆460Updated 2 years ago