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.
☆63Updated 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 code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆48Updated last year
- The official github repo for "Test-Time Training with Masked Autoencoders"☆80Updated last year
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆80Updated last year
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆82Updated last year
- Transformers w/o Attention, based fully on MLPs☆93Updated 11 months ago
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆173Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning☆41Updated 11 months ago
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆60Updated 11 months ago
- Visualizing representations with diffusion based conditional generative model.☆93Updated last year
- A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.☆18Updated last year
- State Space Models☆67Updated 10 months ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆40Updated last year
- ☆42Updated last year
- PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"☆48Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆125Updated 3 years ago
- (ICLR 2023) Official PyTorch implementation of "What Do Self-Supervised Vision Transformers Learn?"☆106Updated last year
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆58Updated last year
- ☆42Updated 2 years ago
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆28Updated 5 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.☆26Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆55Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆106Updated 5 months ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆31Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆81Updated last month
- ☆127Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆150Updated last month