Alternatives and similar repositories for fast_pytorch_kmeans:

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

• subhadarship / kmeans_pytorch
  kmeans using PyTorch
  ☆513Updated last year
• DeMoriarty / TorchPQ
  Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
  ☆223Updated last year
• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆426Updated 6 months ago
• ssnl / align_uniform
  Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…
  ☆441Updated 2 years ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆581Updated 6 months 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.
  ☆64Updated 2 years ago
• AndrewAtanov / simclr-pytorch
  PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results
  ☆203Updated 11 months ago
• salesforce / PCL
  PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
  ☆583Updated 2 years ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆571Updated 4 months ago
• ldeecke / gmm-torch
  Gaussian mixture models in PyTorch.
  ☆555Updated last year
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆458Updated 2 years ago
• Linear95 / CLUB
  Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information
  ☆329Updated 11 months ago
• krasserm / perceiver-io
  A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
  ☆462Updated last year
• janfreyberg / pytorch-revgrad
  A minimal pytorch package implementing a gradient reversal layer.
  ☆158Updated 5 months ago
• lucidrains / slot-attention
  Implementation of Slot Attention from GoogleAI
  ☆425Updated 8 months ago
• joshr17 / HCL
  ICLR 2021, Contrastive Learning with Hard Negative Samples
  ☆265Updated 3 years ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆443Updated 3 months ago
• WeiChengTseng / Pytorch-PCGrad
  Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
  ☆337Updated 3 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆344Updated 5 years ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆457Updated 2 years ago
• Felix-Petersen / difftopk
  Differentiable Top-k Classification Learning
  ☆81Updated 2 years ago
• google-research / diffstride
  TF/Keras code for DiffStride, a pooling layer with learnable strides.
  ☆124Updated 3 years ago
• mxl1990 / tsne-pytorch
  Pytorch implementation for t-SNE with cuda to accelerate
  ☆332Updated 2 years ago
• Spijkervet / SimCLR
  PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
  ☆786Updated 11 months ago
• yukimasano / self-label
  Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)
  ☆538Updated 9 months ago
• ctlllll / SGConv
  ☆164Updated 2 years ago
• chingyaoc / DCL
  NeurIPS 2020, Debiased Contrastive Learning
  ☆282Updated 2 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆279Updated last year
• EPFL-VILAB / MultiMAE
  MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022
  ☆572Updated 2 years ago
• gtegner / mine-pytorch
  Mutual Information Neural Estimation in Pytorch
  ☆319Updated 6 months ago