RakitinDen / pytorch-recursive-gumbel-max-trick
Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021
☆11Updated 2 years ago
Related projects: ⓘ
- Implementation of ICML 22 Paper: Scaling Structured Inference with Randomization☆14Updated 2 years ago
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆32Updated 2 years ago
- ☆12Updated 2 years ago
- Blog post☆16Updated 7 months ago
- Recursive Bayesian Networks☆11Updated last week
- Official PyTorch (Lightning) implementation of the NeurIPS 2020 paper "Efficient Marginalization of Discrete and Structured Latent Variab…☆28Updated 3 years ago
- Implementation of Cascaded Head-colliding Attention (ACL'2021)☆11Updated 3 years ago
- Semiring programming in PyTorch☆10Updated 7 months ago
- Efficient PScan implementation in PyTorch☆15Updated 8 months ago
- ☆21Updated 2 years ago
- Implementation of Recurrent Linear Transformers in Jax+Flax.☆14Updated 10 months ago
- Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)☆10Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆17Updated this week
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated 10 months ago
- Code Repository for "Efficient Computation of Expectations under Spanning Tree Distributions", http://arxiv.org/abs/2008.12988☆10Updated 3 years ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Updated 4 years ago
- A method for evaluating the high-level coherence of machine-generated texts. Identifies high-level coherence issues in transformer-based …☆11Updated last year
- Group-conditional DRO to alleviate spurious correlations☆15Updated 3 years ago
- Code for Residual Energy-Based Models for Text Generation in PyTorch.☆22Updated 3 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆25Updated 2 years ago
- A probabilitic model for contextual word representation. Accepted to ACL2023 Findings.☆16Updated 10 months ago
- ☆32Updated 3 years ago
- Parallel Associative Scan for Language Models☆16Updated 8 months ago
- Suite of 500 procedurally-generated NLP tasks to study language model adaptability☆21Updated 2 years ago
- ☆12Updated 8 months ago
- lanmt ebm☆11Updated 4 years ago
- Code for the paper "Understanding the Mechanics of SPIGOT: Surrogate Gradients for Latent Structure Learning"☆11Updated 3 years ago
- ☆30Updated 2 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Updated 6 years ago
- ☆30Updated 8 months ago