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.
☆65Updated 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:
- The official github repo for "Test-Time Training with Masked Autoencoders"☆82Updated last year
- The code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆51Updated last year
- This is a pytorch implementation of k-means clustering algorithm☆309Updated 2 months ago
- ☆149Updated 2 years ago
- ☆129Updated 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
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆32Updated 2 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆227Updated 2 years ago
- Visualizing representations with diffusion based conditional generative model.☆95Updated 2 years ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆93Updated last year
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- ☆45Updated 2 years ago
- PyTorch Reimplementation of LoRA (featuring with supporting nn.MultiheadAttention)☆61Updated 5 months ago
- Transformers w/o Attention, based fully on MLPs☆93Updated last year
- (ICLR 2023) Official PyTorch implementation of "What Do Self-Supervised Vision Transformers Learn?"☆109Updated last year
- List of papers that combine self-supervision and continual learning☆65Updated 2 months ago
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆60Updated last year
- A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.☆18Updated last year
- [ICML23] On Pitfalls of Test-Time Adaptation☆115Updated last year
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆33Updated 6 months ago
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆46Updated 2 years ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆79Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆143Updated 3 years ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆58Updated last year
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆30Updated 6 months ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆89Updated last year
- Keypoint-Guided Optimal Transport (NeurIPS 2022)☆24Updated last year
- This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.☆27Updated last year
- ResMLP: Feedforward networks for image classification with data-efficient training☆42Updated 4 years ago