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:
- Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)☆36Updated 4 years ago
- Implementation of LogAvgExp for Pytorch☆36Updated 2 months ago
- [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)☆34Updated last year
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆75Updated 2 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆87Updated 4 years ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- ☆89Updated last year
- ☆41Updated 2 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- ☆61Updated 2 years ago
- ☆47Updated 2 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
- ☆33Updated 2 years ago
- ☆51Updated 4 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
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆54Updated 2 years ago
- Code for ICLR 2022 Paper, "Controlling Directions Orthogonal to a Classifier"☆35Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- NeurIPS23 "Flow Factorized Representation Learning"☆36Updated 8 months ago
- ☆51Updated last year
- simple bibtex generator for any text with \cite{}☆31Updated 11 months ago
- Beyond Straight-Through☆100Updated 2 years ago
- Differentiable Top-k Classification Learning☆82Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆91Updated 3 years ago
- Improved diffusion generative models with subspaces☆133Updated 3 years ago
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- Multiscale de-noising score matching☆20Updated 4 years ago