niopeng / dciknn_cuda
This is the CUDA GPU implementation + Python interface (using PyTorch) of DCI. The paper can be found at https://arxiv.org/abs/1512.00442.
☆12Updated last year
Alternatives and similar repositories for dciknn_cuda:
Users that are interested in dciknn_cuda are comparing it to the libraries listed below
- ☆41Updated 2 years ago
- ☆51Updated 9 months ago
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆34Updated 3 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆59Updated 2 years ago
- Code for ICLR 2022 Paper, "Controlling Directions Orthogonal to a Classifier"☆35Updated last year
- Implementation of LogAvgExp for Pytorch☆34Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆73Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆54Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated last year
- ☆50Updated 4 years ago
- ☆32Updated 3 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆85Updated 3 years ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- Improved diffusion generative models with subspaces☆133Updated 2 years ago
- ☆30Updated 4 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆74Updated 2 years ago
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆46Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 5 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Pytorch implementation of The ICML 2020 paper "On Learning Sets of Symmetric Elements" by Haggai Maron, Or Litany, Gal Chechik, Ethan Fet…☆11Updated 3 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- Beyond Straight-Through☆94Updated last year
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields☆67Updated 2 years ago
- ☆61Updated 2 years ago
- ☆88Updated last year
- Weighted Training for Cross-Task Learning☆15Updated 2 years ago