niopeng / dciknn_cudaLinks
This is the CUDA GPU implementation + Python interface (using PyTorch) of DCI. The paper can be found at https://arxiv.org/abs/1512.00442.
☆13Updated 2 years ago
Alternatives and similar repositories for dciknn_cuda
Users that are interested in dciknn_cuda are comparing it to the libraries listed below
Sorting:
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 3 years ago
- Implementation of LogAvgExp for Pytorch☆37Updated 9 months ago
- Differentiable Top-k Classification Learning☆92Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆37Updated 4 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆102Updated 2 years ago
- Multiscale de-noising score matching☆20Updated 4 years ago
- ☆47Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆98Updated 4 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆91Updated 4 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- ☆42Updated 2 years ago
- Understanding the Difficulty of Training Transformers☆47Updated 3 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆62Updated 3 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆57Updated 3 years ago
- Beyond Straight-Through☆108Updated 2 years ago
- Transformers with doubly stochastic attention☆53Updated 3 years ago
- ☆32Updated 5 years ago
- Code for Neural Manifold Clustering and Embedding☆62Updated 3 years ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated 2 years ago
- Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆90Updated 5 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆29Updated 4 years ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆201Updated 2 years ago
- Meta Optimal Transport☆106Updated 2 years ago
- simple bibtex generator for any text with \cite{}☆31Updated last year
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆145Updated 10 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 3 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆231Updated 2 years ago