zugexiaodui / torch_flops
A library for calculating the FLOPs in the forward() process based on torch.fx
☆96Updated 5 months ago
Alternatives and similar repositories for torch_flops:
Users that are interested in torch_flops are comparing it to the libraries listed below
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆248Updated last year
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆393Updated 2 months ago
- ☆39Updated last year
- [CVPR'23] SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformer☆62Updated 9 months ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆79Updated 11 months ago
- Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)☆129Updated last year
- The official implementation of the NeurIPS 2022 paper Q-ViT.☆86Updated last year
- [NeurIPS 2023] Structural Pruning for Diffusion Models☆177Updated 7 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆58Updated 8 months ago
- ☆187Updated last year
- Recent Advances on Efficient Vision Transformers☆49Updated 2 years ago
- The official implementation of the ICML 2023 paper OFQ-ViT☆30Updated last year
- Awesome list of papers that extend Mamba to various applications.☆131Updated 2 months ago
- [TMLR] Official PyTorch implementation of paper "Quantization Variation: A New Perspective on Training Transformers with Low-Bit Precisio…☆41Updated 4 months ago
- [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation☆55Updated last week
- PyTorch code for our paper "ARB-LLM: Alternating Refined Binarizations for Large Language Models"☆22Updated this week
- [ICCV 23]An approach to enhance the efficiency of Vision Transformer (ViT) by concurrently employing token pruning and token merging tech…☆93Updated last year
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Updated 2 years ago
- ☆144Updated last month
- [ICML 2023] This project is the official implementation of our accepted ICML 2023 paper BiBench: Benchmarking and Analyzing Network Binar…☆54Updated 11 months ago
- A repository for DenseSSMs☆86Updated 10 months ago
- ☆174Updated 4 months ago
- [CVPR 2024] PTQ4SAM: Post-Training Quantization for Segment Anything☆62Updated 7 months ago
- PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)☆75Updated last year
- PyTorch implementation of PTQ4DiT https://arxiv.org/abs/2405.16005☆23Updated 3 months ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆296Updated last month
- Join the High Accuracy Club on ImageNet with A Binary Neural Network Ticket☆65Updated 2 years ago
- Curated list of methods that focuses on improving the efficiency of diffusion models☆36Updated 7 months ago
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆283Updated 2 months ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆79Updated last year