Alternatives and similar repositories for pytorch-recursive-gumbel-max-trick:

Users that are interested in pytorch-recursive-gumbel-max-trick are comparing it to the libraries listed below

• xuanlinli17 / autoregressive_inference
  Code for "Discovering Non-monotonic Autoregressive Orderings with Variational Inference" (paper and code updated from ICLR 2021)
  ☆12Updated last year
• thjashin / rodeo
  Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)
  ☆17Updated last year
• bergen / EdgeTransformer
  ☆22Updated 3 years ago
• deep-spin / sparse-communication
  ☆12Updated 3 years ago
• FranxYao / RDP
  Implementation of ICML 22 Paper: Scaling Structured Inference with Randomization
  ☆14Updated 2 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• RobertCsordas / ndr
  The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".
  ☆33Updated 3 years ago
• robert-lieck / RBN
  Recursive Bayesian Networks
  ☆11Updated 4 months ago
• zomux / lanmt-ebm
  lanmt ebm
  ☆11Updated 4 years ago
• nshepperd / gumbel-rao-pytorch
  ☆10Updated 3 years ago
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• maximzubkov / fft-scan
  Efficient PScan implementation in PyTorch
  ☆16Updated last year
• ischlag / Fast-Weight-Memory-public
  Official code repository of the paper Learning Associative Inference Using Fast Weight Memory by Schlag et al.
  ☆28Updated 4 years ago
• deep-spin / sparse-marginalization-lvm
  Official PyTorch (Lightning) implementation of the NeurIPS 2020 paper "Efficient Marginalization of Discrete and Structured Latent Variab…
  ☆28Updated 3 years ago
• AndPotap / einsum-search
  ☆32Updated 6 months ago
• proger / nanokitchen
  Parallel Associative Scan for Language Models
  ☆18Updated last year
• agadetsky / pytorch-pl-variance-reduction
  [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution
  ☆37Updated 4 years ago
• Noahs-ARK / RFA
  ☆33Updated 4 years ago
• RobertCsordas / linear_layer_as_attention
  The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …
  ☆16Updated last year
• subho406 / agalite
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆18Updated 6 months ago
• sustcsonglin / gated_linear_attention_layer
  ☆32Updated last year
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• yikangshen / megablocks
  ☆20Updated 10 months ago
• sustcsonglin / mamba-triton
  ☆46Updated last year
• rycolab / aflt-f2023
  Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)
  ☆10Updated 2 years ago
• Noahs-ARK / PaLM
  PyTorch implementation for PaLM: A Hybrid Parser and Language Model.
  ☆10Updated 5 years ago
• ARM-gradient / ARSM
  Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…
  ☆17Updated 4 years ago
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆26Updated last year
• radarFudan / Curse-of-memory
  Curse-of-memory phenomenon of RNNs in sequence modelling
  ☆19Updated 2 weeks ago
• jenni-ai / T2FW
  Fine-Tuning Pre-trained Transformers into Decaying Fast Weights
  ☆19Updated 2 years ago