Alternatives and similar repositories for torch_flops

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

• transformer-vq / transformer_vq
  ☆200Updated 2 years ago
• kyegomez / SwitchTransformers
  Implementation of Switch Transformers from the paper: "Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficien…
  ☆135Updated this week
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆421Updated 4 months ago
• tensorgi / TPA
  [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)
  ☆443Updated 3 weeks ago
• attention-survey / Efficient_Attention_Survey
  A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention
  ☆270Updated last month
• MzeroMiko / mamba-mini
  An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…
  ☆101Updated 3 months ago
• Dao-AILab / causal-conv1d
  Causal depthwise conv1d in CUDA, with a PyTorch interface
  ☆698Updated this week
• 42Shawn / PTQ4DM
  Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)
  ☆140Updated 2 years ago
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆614Updated 2 months ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆263Updated 6 months ago
• qiuzh20 / gated_attention
  The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…
  ☆772Updated 3 weeks ago
• mit-han-lab / sparsevit
  [CVPR'23] SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformer
  ☆80Updated last year
• A-suozhang / Awesome-Efficient-Diffusion
  Curated list of methods that focuses on improving the efficiency of diffusion models
  ☆44Updated last year
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆397Updated 3 months ago
• ThisisBillhe / EfficientDM
  [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…
  ☆66Updated last year
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆338Updated 10 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆138Updated 7 months ago
• MrYxJ / calculate-flops.pytorch
  The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer…
  ☆911Updated last year
• thu-nics / ViDiT-Q
  [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation
  ☆145Updated 9 months ago
• VainF / Diff-Pruning
  [NeurIPS 2023] Structural Pruning for Diffusion Models
  ☆215Updated last year
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆335Updated 2 months ago
• Efficient-ML / Awesome-Efficient-AIGC
  A list of papers, docs, codes about efficient AIGC. This repo is aimed to provide the info for efficient AIGC research, including languag…
  ☆203Updated 11 months ago
• ziplab / SN-Net
  [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".
  ☆250Updated 2 years ago
• Juanerx / Q-DiT
  [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆73Updated last year
• HuangOwen / Quantization-Variation
  [TMLR] Official PyTorch implementation of paper "Quantization Variation: A New Perspective on Training Transformers with Low-Bit Precisio…
  ☆46Updated last year
• OpenGVLab / Vision-RWKV
  [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures
  ☆536Updated 10 months ago
• VainF / MaskLLM-4V
  Learnable Semi-structured Sparsity for Vision Transformers and Diffusion Transformers
  ☆14Updated 11 months ago
• tommyip / mamba2-minimal
  Minimal Mamba-2 implementation in PyTorch
  ☆240Updated last year
• Hao840 / Awesome-Low-Precision-Training
  A collection of research papers on low-precision training methods
  ☆57Updated 8 months ago
• RobinWu218 / ToST
  [ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
  ☆129Updated 10 months ago