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 last year
Alternatives and similar repositories for dciknn_cuda
Users that are interested in dciknn_cuda are comparing it to the libraries listed below
Sorting:
- ☆42Updated 2 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆62Updated 3 years ago
- ☆52Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Beyond Straight-Through☆103Updated 2 years ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆36Updated 4 years ago
- Implementation of LogAvgExp for Pytorch☆37Updated 7 months ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- Code for ICLR 2022 Paper, "Controlling Directions Orthogonal to a Classifier"☆35Updated 2 years ago
- ☆53Updated 4 years ago
- Differentiable Top-k Classification Learning☆87Updated 2 years ago
- ☆32Updated 4 years ago
- Natural Language Descriptions of Deep Visual Features, ICLR 2022☆64Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆66Updated 3 years ago
- implements optimal transport algorithms in pytorch☆102Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆96Updated 3 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆67Updated 4 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 for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆89Updated 4 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Updated 3 years ago
- Standalone Product Key Memory module in Pytorch - for augmenting Transformer models☆85Updated 2 weeks ago
- Unofficial PyTorch implementation of "Step-unrolled Denoising Autoencoders for Text Generation"☆24Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 4 years ago
- ☆38Updated 4 years ago
- Transformers with doubly stochastic attention☆50Updated 3 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆46Updated 2 years ago