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:
- ☆51Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- Implementation of LogAvgExp for Pytorch☆36Updated 4 months ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆36Updated 4 years ago
- Beyond Straight-Through☆100Updated 2 years ago
- ☆41Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆62Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 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
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆76Updated 2 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆57Updated 2 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆90Updated 4 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆92Updated 3 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 2 years ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆197Updated last year
- ☆61Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆155Updated last year
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Updated 3 years ago
- ☆32Updated 4 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆68Updated 4 years ago
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- ☆52Updated 4 years ago
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆75Updated 2 years ago
- Meta Optimal Transport☆103Updated 2 years ago
- Natural Language Descriptions of Deep Visual Features, ICLR 2022☆65Updated 2 years ago
- ☆229Updated 2 years ago
- Standalone Product Key Memory module in Pytorch - for augmenting Transformer models☆82Updated last year
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆71Updated last year
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated 2 years ago