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(Bert、LlaMA etc Large Language Model)
☆761Updated 9 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
- Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4☆803Updated this week
- [TMLR 2024] Efficient Large Language Models: A Survey☆1,140Updated 3 weeks ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆448Updated 7 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,002Updated 6 months ago
- 📰 Must-read papers and blogs on Speculative Decoding ⚡️☆696Updated last week
- List of papers related to neural network quantization in recent AI conferences and journals.☆597Updated 3 weeks ago
- ☆237Updated 8 months ago
- Fast inference from large lauguage models via speculative decoding☆714Updated 8 months ago
- Official Implementation of EAGLE-1 (ICML'24), EAGLE-2 (EMNLP'24), and EAGLE-3.☆1,197Updated last week
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆730Updated last year
- Awesome list for LLM pruning.☆222Updated 4 months ago
- A curated reading list of research in Mixture-of-Experts(MoE).☆617Updated 5 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆598Updated last year
- ☆313Updated last year
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆165Updated 11 months ago
- Collection of papers on state-space models☆588Updated last week
- A simple and effective LLM pruning approach.☆738Updated 8 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton☆2,305Updated this week
- A library for calculating the FLOPs in the forward() process based on torch.fx☆104Updated 3 weeks ago
- Pipeline Parallelism for PyTorch☆764Updated 8 months ago
- Survey Paper List - Efficient LLM and Foundation Models☆246Updated 7 months ago
- [ICML2024 (Oral)] Official PyTorch implementation of DoRA: Weight-Decomposed Low-Rank Adaptation☆766Updated 6 months ago
- Helpful tools and examples for working with flex-attention☆726Updated 2 weeks ago
- Ring attention implementation with flash attention☆743Updated 2 weeks ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆604Updated 11 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆477Updated this week
- Awesome LLM compression research papers and tools.☆1,472Updated last week
- A collection of AWESOME things about mixture-of-experts☆1,087Updated 4 months ago
- A curated list for Efficient Large Language Models☆1,614Updated last week
- The official GitHub page for the survey paper "A Survey on Mixture of Experts in Large Language Models".☆331Updated last month