Alternatives and similar repositories for rmsnorm:

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

• fkodom / grouped-query-attention-pytorch
  (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …
  ☆162Updated 11 months ago
• ofirpress / attention_with_linear_biases
  Code for the ALiBi method for transformer language models (ICLR 2022)
  ☆521Updated last year
• facebookresearch / mega
  Sequence modeling with Mega.
  ☆295Updated 2 years ago
• lucidrains / CoLT5-attention
  Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch
  ☆226Updated 7 months ago
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆278Updated last month
• booydar / LM-RMT
  Recurrent Memory Transformer
  ☆149Updated last year
• princeton-nlp / CoFiPruning
  [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408
  ☆195Updated last year
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆279Updated last year
• OpenNLPLab / TransnormerLLM
  Official implementation of TransNormerLLM: A Faster and Better LLM
  ☆243Updated last year
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆225Updated 3 years ago
• lucidrains / memory-efficient-attention-pytorch
  Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
  ☆376Updated last year
• haoliuhl / ringattention
  Large Context Attention
  ☆703Updated 2 months ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆213Updated 2 years ago
• lucidrains / st-moe-pytorch
  Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch
  ☆328Updated 10 months ago
• bobby-he / simplified_transformers
  ☆290Updated 4 months ago
• bojone / rerope
  Rectified Rotary Position Embeddings
  ☆365Updated 11 months ago
• lucidrains / recurrent-memory-transformer-pytorch
  Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch
  ☆407Updated 3 months ago
• epfml / dynamic-sparse-flash-attention
  ☆143Updated last year
• yxli2123 / LoftQ
  ☆219Updated 10 months ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆263Updated 3 years ago
• haochengxi / Train_Transformers_with_INT4
  ☆146Updated last year
• HazyResearch / fly
  ☆202Updated 2 years ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆440Updated 3 months ago
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4
  ☆800Updated this week
• berlino / gated_linear_attention
  ☆102Updated last year
• google-deepmind / randomized_positional_encodings
  Randomized Positional Encodings Boost Length Generalization of Transformers
  ☆80Updated last year
• OpenNLPLab / cosFormer
  [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention
  ☆190Updated 2 years ago
• facebookresearch / LLM-QAT
  Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"
  ☆278Updated last month
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆665Updated 4 months ago
• kyegomez / FlashAttention20
  Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels
  ☆102Updated last year