ShishirPatil / poet
ML model training for edge devices
☆160Updated last year
Alternatives and similar repositories for poet:
Users that are interested in poet are comparing it to the libraries listed below
- ☆246Updated 6 months ago
- ☆157Updated last year
- AI and Memory Wall☆212Updated 10 months ago
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆47Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- A schedule language for large model training☆144Updated 8 months ago
- Compression for Foundation Models☆24Updated last week
- ☆100Updated 5 months ago
- Advanced Ultra-Low Bitrate Compression Techniques for the LLaMA Family of LLMs☆111Updated last year
- vLLM: A high-throughput and memory-efficient inference and serving engine for LLMs☆88Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆154Updated 2 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- ☆143Updated last year
- Reorder-based post-training quantization for large language model☆184Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆221Updated 6 months ago
- PB-LLM: Partially Binarized Large Language Models☆151Updated last year
- ☆92Updated 2 years ago
- Official code for "SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient"☆137Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆199Updated last year
- ☆141Updated 3 weeks ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆55Updated 3 years ago
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆266Updated last year
- pytorch-profiler☆50Updated last year
- ☆59Updated 2 weeks ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆58Updated last month
- A library to analyze PyTorch traces.☆332Updated last week
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆229Updated this week
- Torch Distributed Experimental☆115Updated 6 months ago
- Applied AI experiments and examples for PyTorch☆225Updated this week
- ☆43Updated 7 months ago