idiap / fast-transformers

Related projects ⓘ

Alternatives and complementary repositories for fast-transformers

• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,098Updated 2 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆698Updated 6 months ago
• lucidrains / reformer-pytorch
  Reformer, the efficient Transformer, in Pytorch
  ☆2,121Updated last year
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆729Updated 11 months ago
• tatp22 / linformer-pytorch
  My take on a practical implementation of Linformer for Pytorch.
  ☆407Updated 2 years ago
• allenai / longformer
  Longformer: The Long-Document Transformer
  ☆2,047Updated last year
• openai / sparse_attention
  Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"
  ☆1,524Updated 4 years ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,097Updated last year
• sacmehta / delight
  DeLighT: Very Deep and Light-Weight Transformers
  ☆467Updated 4 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,020Updated 3 months ago
• lucidrains / routing-transformer
  Fully featured implementation of Routing Transformer
  ☆284Updated 3 years ago
• jettify / pytorch-optimizer
  torch-optimizer -- collection of optimizers for Pytorch
  ☆3,043Updated 7 months ago
• davidmrau / mixture-of-experts
  PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538
  ☆983Updated 7 months ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆384Updated 2 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆571Updated last week
• mit-han-lab / lite-transformer
  [ICLR 2020] Lite Transformer with Long-Short Range Attention
  ☆598Updated 4 months ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,003Updated 2 months ago
• lucidrains / mixture-of-experts
  A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models
  ☆642Updated last year
• state-spaces / s4
  Structured state space sequence models
  ☆2,470Updated 4 months ago
• ofirpress / attention_with_linear_biases
  Code for the ALiBi method for transformer language models (ICLR 2022)
  ☆507Updated last year
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆257Updated 10 months ago
• lucidrains / sinkhorn-transformer
  Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention
  ☆253Updated 3 years ago
• facebookresearch / higher
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,593Updated 2 years ago
• harvardnlp / pytorch-struct
  Fast, general, and tested differentiable structured prediction in PyTorch
  ☆1,108Updated 2 years ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,062Updated 2 years ago
• kimiyoung / transformer-xl
  ☆3,612Updated 2 years ago
• Lightning-Universe / lightning-transformers
  Flexible components pairing 🤗 Transformers with Pytorch Lightning
  ☆611Updated last year
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆369Updated 3 years ago
• lucidrains / g-mlp-pytorch
  Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch
  ☆425Updated 3 years ago
• mlpen / Nystromformer
  ☆365Updated last year