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,093Updated 2 years ago
• lucidrains / reformer-pytorch
  Reformer, the efficient Transformer, in Pytorch
  ☆2,116Updated last year
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆695Updated 6 months ago
• tatp22 / linformer-pytorch
  My take on a practical implementation of Linformer for Pytorch.
  ☆407Updated 2 years ago
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆727Updated 10 months ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,092Updated last year
• state-spaces / s4
  Structured state space sequence models
  ☆2,455Updated 3 months ago
• allenai / longformer
  Longformer: The Long-Document Transformer
  ☆2,046Updated last year
• openai / sparse_attention
  Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"
  ☆1,524Updated 4 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,001Updated last month
• jettify / pytorch-optimizer
  torch-optimizer -- collection of optimizers for Pytorch
  ☆3,040Updated 7 months ago
• sacmehta / delight
  DeLighT: Very Deep and Light-Weight Transformers
  ☆466Updated 4 years ago
• rusty1s / pytorch_sparse
  PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
  ☆1,009Updated 2 months ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,018Updated 3 months ago
• rusty1s / pytorch_scatter
  PyTorch Extension Library of Optimized Scatter Operations
  ☆1,557Updated 2 months 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,589Updated 2 years ago
• lucidrains / routing-transformer
  Fully featured implementation of Routing Transformer
  ☆284Updated 3 years 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
  ☆974Updated 6 months ago
• Lightning-AI / deep-learning-project-template
  Pytorch Lightning code guideline for conferences
  ☆1,240Updated last year
• lucidrains / g-mlp-pytorch
  Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch
  ☆423Updated 3 years ago
• harvardnlp / pytorch-struct
  Fast, general, and tested differentiable structured prediction in PyTorch
  ☆1,108Updated 2 years ago
• Lightning-Universe / lightning-bolts
  Toolbox of models, callbacks, and datasets for AI/ML researchers.
  ☆1,691Updated this week
• juntang-zhuang / Adabelief-Optimizer
  Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"
  ☆1,050Updated 3 months ago
• lucidrains / lambda-networks
  Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute
  ☆1,532Updated 3 years ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆727Updated 2 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆545Updated 3 years ago
• mit-han-lab / lite-transformer
  [ICLR 2020] Lite Transformer with Long-Short Range Attention
  ☆599Updated 3 months ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆383Updated 2 months ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆759Updated 4 months ago
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆369Updated 3 years ago