Alternatives and similar repositories for torch_kmeans

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

• VITA-Group / ViT-Anti-Oversmoothing
  [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…
  ☆80Updated last year
• DeMoriarty / fast_pytorch_kmeans
  This is a pytorch implementation of k-means clustering algorithm
  ☆321Updated 5 months ago
• PKU-ML / non_neg
  Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
  ☆45Updated last year
• fawazsammani / awesome-mlp-mixer
  Transformers w/o Attention, based fully on MLPs
  ☆94Updated last year
• needylove / OrdinalEntropy
  The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023
  ☆51Updated last year
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆97Updated 2 years ago
• yossigandelsman / test_time_training_mae
  The official github repo for "Test-Time Training with Masked Autoencoders"
  ☆87Updated last year
• quanlin-wu / dmae
  Denoising Masked Autoencoders Help Robust Classification.
  ☆67Updated 2 years ago
• xxgege / GAM
  The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".
  ☆85Updated 2 years ago
• lucidrains / bidirectional-cross-attention
  A simple cross attention that updates both the source and target in one step
  ☆176Updated 3 weeks ago
• AvivNavon / DWSNets
  Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]
  ☆89Updated 2 years ago
• sndnyang / Diffusion_ViT
  PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"
  ☆49Updated 2 years ago
• zhangq327 / U-MAE
  Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders
  ☆68Updated last year
• XzwHan / CARD
  Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"
  ☆230Updated 2 years ago
• nblt / F-SAM
  [CVPR 2024] Friendly Sharpness-Aware Minimization
  ☆34Updated 9 months ago
• Felix-Petersen / difftopk
  Differentiable Top-k Classification Learning
  ☆85Updated 2 years ago
• maxvanspengler / hyperbolic_learning_library
  An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.
  ☆166Updated 7 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆100Updated last week
• KingJamesSong / DifferentiableSVD
  A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…
  ☆78Updated last year
• willGuimont / learnable_fourier_positional_encoding
  Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding
  ☆55Updated 10 months ago
• AlexGraikos / diffusion_priors
  Using pre-trained Diffusion models as priors for inference tasks
  ☆208Updated 2 years ago
• badripatro / mamba360
  State Space Models
  ☆70Updated last year
• alihassanijr / TorchKMeans
  A torch-based implementation of K-Means and K-Means++
  ☆17Updated 4 years ago
• ShaharLutatiPersonal / OCD
  Official PyTorch Implementation
  ☆18Updated 2 years ago
• Haochen-Wang409 / DropPos
  [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions
  ☆61Updated last year
• jhaochenz96 / spectral_contrastive_learning
  ☆22Updated 3 years ago
• LeslieTrue / CPP
  This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.
  ☆28Updated 2 years ago
• Hzzone / torch_clustering
  A pure PyTorch implementation of kmeans and GMM with distributed clustering.
  ☆48Updated 2 years ago
• connorlee77 / pytorch-mutual-information
  Mutual Information in Pytorch
  ☆112Updated 2 years ago
• qym7 / CBDM-pytorch
  Official implementation for 'Class-Balancing Diffusion Models'
  ☆54Updated last year