Alternatives and similar repositories for torch_flops

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

• MzeroMiko / mamba-mini
  An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…
  ☆93Updated last year
• transformer-vq / transformer_vq
  ☆195Updated last year
• Dao-AILab / causal-conv1d
  Causal depthwise conv1d in CUDA, with a PyTorch interface
  ☆510Updated 2 weeks ago
• 42Shawn / PTQ4DM
  Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)
  ☆139Updated 2 years ago
• kyegomez / SwitchTransformers
  Implementation of Switch Transformers from the paper: "Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficien…
  ☆110Updated 3 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…
  ☆820Updated last year
• A-suozhang / Awesome-Efficient-Diffusion
  Curated list of methods that focuses on improving the efficiency of diffusion models
  ☆45Updated last year
• VainF / Diff-Pruning
  [NeurIPS 2023] Structural Pruning for Diffusion Models
  ☆198Updated last year
• tensorgi / TPA
  The official implementation of TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)
  ☆376Updated last week
• megvii-research / FQ-ViT
  [IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer
  ☆343Updated 2 years ago
• OpenNLPLab / lightning-attention
  Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models
  ☆322Updated 4 months ago
• ziplab / SN-Net
  [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".
  ☆247Updated 2 years ago
• ghimiredhikura / Awasome-Pruning
  Awasome Papers and Resources in Deep Neural Network Pruning with Source Code.
  ☆160Updated 10 months ago
• mit-han-lab / sparsevit
  [CVPR'23] SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformer
  ☆72Updated last year
• Hsu1023 / DuQuant
  [NeurIPS 2024 Oral🔥] DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs.
  ☆164Updated 9 months ago
• kyegomez / AttentionIsOFFByOne
  Implementation of "Attention Is Off By One" by Evan Miller
  ☆193Updated last year
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆594Updated 2 months ago
• MingSun-Tse / Awesome-Efficient-ViT
  Recent Advances on Efficient Vision Transformers
  ☆51Updated 2 years ago
• YanjingLi0202 / Q-ViT
  The official implementation of the NeurIPS 2022 paper Q-ViT.
  ☆96Updated 2 years ago
• Qualcomm-AI-research / outlier-free-transformers
  ☆42Updated last year
• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆350Updated 6 months ago
• adreamwu / PTQ4DiT
  PyTorch implementation of PTQ4DiT https://arxiv.org/abs/2405.16005
  ☆31Updated 8 months ago
• OpenGVLab / Vision-RWKV
  [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures
  ☆477Updated 4 months ago
• ThisisBillhe / EfficientDM
  [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…
  ☆60Updated last year
• SHI-Labs / NATTEN
  Neighborhood Attention Extension. Bringing attention to a neighborhood near you!
  ☆542Updated last week
• thu-nics / ViDiT-Q
  [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation
  ☆102Updated 3 months ago
• HuangOwen / Quantization-Variation
  [TMLR] Official PyTorch implementation of paper "Quantization Variation: A New Perspective on Training Transformers with Low-Bit Precisio…
  ☆45Updated 9 months ago
• tommyip / mamba2-minimal
  Minimal Mamba-2 implementation in PyTorch
  ☆206Updated last year
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆191Updated last week
• AvivBick / awesome-ssm-ml
  Reading list for research topics in state-space models
  ☆302Updated last month