ShishirPatil / poet
ML model training for edge devices
☆162Updated last year
Alternatives and similar repositories for poet:
Users that are interested in poet are comparing it to the libraries listed below
- Official code for "SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient"☆138Updated last year
- A schedule language for large model training☆146Updated 10 months ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆61Updated 3 months ago
- ☆251Updated 9 months ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 3 years ago
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆49Updated last year
- ☆157Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆206Updated last year
- vLLM: A high-throughput and memory-efficient inference and serving engine for LLMs☆86Updated this week
- ☆141Updated 9 months ago
- AI and Memory Wall☆215Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees"☆363Updated last year
- ☆45Updated 10 months ago
- ☆147Updated last year
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆108Updated 4 months ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆116Updated last year
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆305Updated 9 months ago
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆687Updated 8 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- ☆103Updated 8 months ago
- ☆143Updated 2 years ago
- Research and development for optimizing transformers☆125Updated 4 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated 2 years ago
- How much energy do GenAI models consume?☆42Updated 6 months ago
- Pytorch distributed backend extension with compression support☆16Updated last month
- Advanced Ultra-Low Bitrate Compression Techniques for the LLaMA Family of LLMs☆111Updated last year
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆38Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆223Updated 8 months ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆403Updated last week