jokofa / torch_kmeansLinks
PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
☆75Updated 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
Sorting:
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆81Updated last year
- Visualizing representations with diffusion based conditional generative model.☆102Updated 2 years ago
- This is a pytorch implementation of k-means clustering algorithm☆333Updated 8 months ago
- Using pre-trained Diffusion models as priors for inference tasks☆208Updated 2 years ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆36Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆84Updated 2 years ago
- This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.☆31Updated 2 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆232Updated 2 years ago
- The official github repo for "Test-Time Training with Masked Autoencoders"☆88Updated 2 years ago
- The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆53Updated 2 years ago
- This repository contains the implementation for the paper "EMP-SSL: Towards Self-Supervised Learning in One Training Epoch."☆230Updated 2 years ago
- PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"☆49Updated 2 years ago
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆58Updated 2 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆78Updated 2 years ago
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆34Updated last year
- ☆142Updated last year
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- A simple cross attention that updates both the source and target in one step☆185Updated 3 months ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆68Updated last year
- [NeurIPS 2023] code for "DisDiff: Unsupervised Disentanglement of Diffusion Probabilistic Models☆71Updated 2 years ago
- ☆65Updated 3 years ago
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆123Updated 7 months ago
- Denoising Masked Autoencoders Help Robust Classification.☆67Updated 2 years ago
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆31Updated last year
- ☆58Updated last year
- Official PyTorch code for "Out-of-distribution detection with denoising diffusion models"☆51Updated last year
- Implementation of various DDPM papers to understand how they work☆89Updated 7 months ago
- PyTorch implementation of D2C: Diffuison-Decoding Models for Few-shot Conditional Generation.☆125Updated 3 years ago
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆63Updated 3 years ago