eamartin / parallelizing_linear_rnns

Related projects: ⓘ

• sustcsonglin / mamba-triton
  ☆42Updated 7 months ago
• proger / nanokitchen
  Parallel Associative Scan for Language Models
  ☆16Updated 8 months ago
• sustcsonglin / gated_linear_attention_layer
  ☆30Updated 8 months ago
• OpenNLPLab / HGRN
  [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…
  ☆60Updated 4 months ago
• ag1988 / dlr
  The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…
  ☆19Updated last year
• berlino / seq_icl
  ☆48Updated 4 months ago
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• srush / mamba-scans
  Blog post
  ☆16Updated 7 months ago
• johnryan465 / pscan
  ☆33Updated 8 months ago
• HazyResearch / prefix-linear-attention
  ☆41Updated 2 months ago
• renll / SeqBoat
  [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling
  ☆34Updated 9 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆40Updated last month
• microsoft / SparseMixer
  Sparse Backpropagation for Mixture-of-Expert Training
  ☆17Updated 2 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆86Updated 3 months ago
• conceptofmind / flash-gpt
  ☆34Updated this week
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆28Updated last week
• maximzubkov / fft-scan
  Efficient PScan implementation in PyTorch
  ☆15Updated 8 months ago
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆66Updated last year
• lucidrains / pause-transformer
  Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…
  ☆50Updated 10 months ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated last year
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆55Updated 4 months ago
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆94Updated last year
• lxxue / prefix_sum
  A PyTorch wrapper of parallel exclusive scan in CUDA
  ☆8Updated last year
• languini-kitchen / languini-kitchen
  The official Languini Kitchen repository
  ☆14Updated 4 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆151Updated last month
• radarFudan / mamba-minimal-jax
  ☆28Updated last week
• mgmalek / efficient_cross_entropy
  ☆66Updated 3 months ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆65Updated 11 months ago
• McGill-NLP / length-generalization
  Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023
  ☆123Updated 4 months ago
• pkuzengqi / Skyformer
  Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)
  ☆58Updated 2 years ago