Alternatives and similar repositories for Tnn

Users that are interested in Tnn are comparing it to the libraries listed below

• OpenNLPLab / Transnormer
  [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformer
  ☆64Updated 2 years ago
• OpenNLPLab / HGRN
  [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…
  ☆66Updated last year
• lsj2408 / URPE
  [NeurIPS 2022] Your Transformer May Not be as Powerful as You Expect (official implementation)
  ☆34Updated 2 years ago
• berlino / gated_linear_attention
  ☆106Updated last year
• google-deepmind / randomized_positional_encodings
  Randomized Positional Encodings Boost Length Generalization of Transformers
  ☆83Updated last year
• HazyResearch / prefix-linear-attention
  ☆57Updated last year
• yegcjs / DiffusionLLM
  Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"
  ☆83Updated last year
• OpenNLPLab / ETSC-Exact-Toeplitz-to-SSM-Conversion
  [EMNLP 2023] Official implementation of the algorithm ETSC: Exact Toeplitz-to-SSM Conversion our EMNLP 2023 paper - Accelerating Toeplitz…
  ☆14Updated 2 years ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆134Updated last week
• VITA-Group / Random-MoE-as-Dropout
  [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…
  ☆56Updated 2 years ago
• igul222 / plaid
  ☆110Updated 2 years ago
• haoliuhl / language-quantized-autoencoders
  Language Quantized AutoEncoders
  ☆111Updated 2 years ago
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆62Updated last year
• YuchuanTian / DiJiang
  [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…
  ☆104Updated last year
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆101Updated 2 years ago
• Doraemonzzz / tnn-pytorch
  ☆20Updated 2 years ago
• yikangshen / MoA
  Mixture of Attention Heads
  ☆51Updated 3 years ago
• lucidrains / memformer
  Implementation of Memformer, a Memory-augmented Transformer, in Pytorch
  ☆126Updated 5 years ago
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆113Updated 11 months ago
• transformer-vq / transformer_vq
  ☆200Updated 2 years ago
• BaohaoLiao / mefts
  [NeurIPS 2023] Make Your Pre-trained Model Reversible: From Parameter to Memory Efficient Fine-Tuning
  ☆33Updated 2 years ago
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆56Updated last year
• amazon-science / masked-diffusion-lm
  Official implementation for the paper "A Cheaper and Better Diffusion Language Model with Soft-Masked Noise"
  ☆59Updated 2 years ago
• kyegomez / Blockwise-Parallel-Transformer
  32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.
  ☆49Updated 2 years ago
• deep-spin / infinite-former
  ☆67Updated last year
• chijames / KERPLE
  ☆20Updated 3 years ago
• PiotrNawrot / dynamic-pooling
  Efficient Transformers with Dynamic Token Pooling
  ☆66Updated 2 years ago
• chuanyang-Zheng / DAPE
  The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"
  ☆39Updated last year
• shaochenze / PatchTrain
  Code for paper "Patch-Level Training for Large Language Models"
  ☆96Updated last month
• microsoft / EfficientLongSequenceModeling
  ☆52Updated 2 years ago