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.
☆80Updated 3 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 code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆54Updated 2 years ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆83Updated 2 years ago
- This is a pytorch implementation of k-means clustering algorithm☆339Updated 11 months ago
- The official github repo for "Test-Time Training with Masked Autoencoders"☆92Updated 2 years ago
- Visualizing representations with diffusion based conditional generative model.☆105Updated 2 years ago
- ☆65Updated 3 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆79Updated 2 years ago
- Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning☆47Updated last year
- Modular and intuitive Hypernetworks in Pytorch☆40Updated 2 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆234Updated 2 years ago
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆176Updated 2 years ago
- Using pre-trained Diffusion models as priors for inference tasks☆210Updated 2 years ago
- Transformers w/o Attention, based fully on MLPs☆97Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆55Updated last year
- This repository contains the implementation for the paper "EMP-SSL: Towards Self-Supervised Learning in One Training Epoch."☆227Updated 2 years ago
- Official PyTorch code for "Out-of-distribution detection with denoising diffusion models"☆51Updated last year
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆64Updated 3 years ago
- Mutual Information in Pytorch☆115Updated 2 years ago
- Official PyTorch Implementation☆17Updated 3 years ago
- A simple cross attention that updates both the source and target in one step☆195Updated 6 months ago
- Hyperbolic Image Segmentation, CVPR 2022☆93Updated 3 years ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆68Updated 2 years ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆37Updated 2 years ago
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆50Updated 3 years ago
- An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.☆176Updated last year
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆42Updated 2 years ago
- B-cos Networks: Alignment is All we Need for Interpretability☆114Updated 2 years ago
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆131Updated 9 months ago
- PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"☆49Updated 3 years ago