DeMoriarty / fast_pytorch_kmeansLinks
This is a pytorch implementation of k-means clustering algorithm
☆321Updated 5 months ago
Alternatives and similar repositories for fast_pytorch_kmeans
Users that are interested in fast_pytorch_kmeans are comparing it to the libraries listed below
Sorting:
- kmeans using PyTorch☆515Updated 2 years ago
- Implementation of Linformer for Pytorch☆295Updated last year
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆225Updated last year
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆69Updated 2 years ago
- An implementation of local windowed attention for language modeling☆472Updated last month
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆435Updated 10 months ago
- Differentiable Top-k Classification Learning☆85Updated 2 years ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆603Updated 10 months ago
- ☆138Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆601Updated 8 months ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆360Updated 4 years ago
- Gaussian mixture models in PyTorch.☆570Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆253Updated 4 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆196Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆487Updated last year
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆267Updated 4 years ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆448Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆794Updated last year
- ☆164Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆736Updated 3 weeks ago
- Learning Rate Warmup in PyTorch☆411Updated 2 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)☆144Updated 3 years ago
- PyTorch implementation of HyperNetworks (Ha et al., ICLR 2017) for ResNet (Residual Networks)☆269Updated 4 years ago
- An implementation of the efficient attention module.☆320Updated 4 years ago
- Unofficial PyTorch implementation of Google's FNet: Mixing Tokens with Fourier Transforms. With checkpoints.☆74Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated 2 years ago
- A simple cross attention that updates both the source and target in one step☆176Updated 3 weeks ago
- Compare neural networks by their feature similarity☆370Updated 2 years ago