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.
☆67Updated 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 code of "Improving Deep Regression with Ordinal Entropy" in ICLR 2023☆51Updated last year
- This is a pytorch implementation of k-means clustering algorithm☆314Updated 4 months ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆83Updated 2 years ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆80Updated last year
- Mutual Information in Pytorch☆112Updated last year
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆78Updated last year
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆33Updated 8 months ago
- A simple cross attention that updates both the source and target in one step☆172Updated last year
- Visualizing representations with diffusion based conditional generative model.☆95Updated 2 years ago
- This repository contains the implementation for the paper "EMP-SSL: Towards Self-Supervised Learning in One Training Epoch."☆227Updated last year
- Minimal implementation of PCA in PyTorch, tested against scikit-learn's implementation☆28Updated 4 months ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆228Updated 2 years ago
- Hyperbolic Image Segmentation, CVPR 2022☆87Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆89Updated last year
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆67Updated last year
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆33Updated 2 years ago
- Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning☆45Updated last year
- Papers and Codes for the deep learning in hyperbolic space☆177Updated 2 years ago
- Transformers w/o Attention, based fully on MLPs☆93Updated last year
- The official github repo for "Test-Time Training with Masked Autoencoders"☆86Updated last year
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆175Updated 2 years ago
- An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.☆162Updated 6 months 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
- An official codebase of paper "Revisiting Sparse Convolutional Model for Visual Recognition"☆124Updated 2 years ago
- ☆36Updated 2 years ago
- Using pre-trained Diffusion models as priors for inference tasks☆207Updated 2 years ago
- Compare neural networks by their feature similarity☆365Updated 2 years ago
- State Space Models☆68Updated last year
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆47Updated 2 years ago
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆30Updated 8 months ago