MrYxJ / calculate-flops.pytorchLinks
The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)
☆797Updated 11 months ago
Alternatives and similar repositories for calculate-flops.pytorch
Users that are interested in calculate-flops.pytorch are comparing it to the libraries listed below
Sorting:
- Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4☆829Updated this week
- A collection of AWESOME things about mixture-of-experts☆1,135Updated 5 months ago
- A curated reading list of research in Mixture-of-Experts(MoE).☆628Updated 7 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton☆2,438Updated last week
- 📰 Must-read papers and blogs on Speculative Decoding ⚡️☆755Updated last week
- Ring attention implementation with flash attention☆771Updated last week
- A library for calculating the FLOPs in the forward() process based on torch.fx☆112Updated 2 months ago
- The official GitHub page for the survey paper "A Survey on Mixture of Experts in Large Language Models".☆361Updated 2 months ago
- Helpful tools and examples for working with flex-attention☆811Updated this week
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆294Updated 3 months ago
- Efficient Multimodal Large Language Models: A Survey☆349Updated last month
- Fast inference from large lauguage models via speculative decoding☆745Updated 9 months ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆471Updated 8 months ago
- depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.☆686Updated last month
- [TMLR 2024] Efficient Large Language Models: A Survey☆1,161Updated 2 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,417Updated 10 months ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆613Updated last year
- 📰 Must-read papers on KV Cache Compression (constantly updating 🤗).☆431Updated last week
- Microsoft Automatic Mixed Precision Library☆602Updated 8 months ago
- ☆638Updated 2 weeks ago
- Survey Paper List - Efficient LLM and Foundation Models☆248Updated 8 months ago
- [NeurIPS 2023] LLM-Pruner: On the Structural Pruning of Large Language Models. Support Llama-3/3.1, Llama-2, LLaMA, BLOOM, Vicuna, Baich…☆1,014Updated 7 months ago
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆686Updated 2 months ago
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆287Updated 3 months ago
- Official Implementation of EAGLE-1 (ICML'24), EAGLE-2 (EMNLP'24), and EAGLE-3.☆1,277Updated this week
- TransMLA: Multi-Head Latent Attention Is All You Need☆284Updated this week
- A curated list for Efficient Large Language Models☆1,694Updated last month
- The official implementation of Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆373Updated 2 weeks ago
- Collection of papers on state-space models☆593Updated last month
- 🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"☆686Updated 2 months ago