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
Alternatives and similar repositories for torch_kmeans:
Users that are interested in torch_kmeans are comparing it to the libraries listed below
- The official github repo for "Test-Time Training with Masked Autoencoders"☆82Updated last year
- Visualizing representations with diffusion based conditional generative model.☆94Updated last year
- Transformers w/o Attention, based fully on MLPs☆93Updated last year
- ☆149Updated 2 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- B-cos Networks: Alignment is All we Need for Interpretability☆109Updated last year
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆173Updated last year
- The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆50Updated last year
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆31Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆77Updated last year
- ☆45Updated 2 years ago
- Official PyTorch code for "Out-of-distribution detection with denoising diffusion models"☆49Updated 10 months ago
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆44Updated 2 years ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆60Updated 11 months ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated 2 years ago
- Decoupled Kullback-Leibler Divergence Loss (DKL), NeurIPS 2024 / Generalized Kullback-Leibler Divergence Loss (GKL)☆44Updated 3 weeks ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆33Updated 5 months ago
- Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning☆44Updated last year
- Pytorch implementation for ICLR 2021 paper - MiCE: Mixture of Contrastive Experts for Unsupervised Image Clustering☆49Updated 4 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆143Updated 2 years ago
- (ICLR 2023) Official PyTorch implementation of "What Do Self-Supervised Vision Transformers Learn?"☆108Updated last year
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Hyperbolic Image Segmentation, CVPR 2022☆85Updated 2 years ago
- Using pre-trained Diffusion models as priors for inference tasks☆201Updated 2 years ago
- Minimal implementation of PCA in PyTorch, tested against scikit-learn's implementation☆28Updated 2 months ago
- List of papers that combine self-supervision and continual learning☆62Updated last month
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆62Updated 2 years ago
- ☆63Updated 2 years ago