intelligent-machine-learning / atorchLinks
An industrial extension library of pytorch to accelerate large scale model training
☆36Updated 2 months ago
Alternatives and similar repositories for atorch
Users that are interested in atorch are comparing it to the libraries listed below
Sorting:
- ☆145Updated 4 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆132Updated 6 months ago
- ☆77Updated 3 months ago
- Allow torch tensor memory to be released and resumed later☆80Updated last week
- FlagScale is a large model toolkit based on open-sourced projects.☆325Updated this week
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆226Updated last week
- ☆90Updated 3 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆215Updated last month
- Zero Bubble Pipeline Parallelism☆406Updated 2 months ago
- ☆128Updated 6 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆102Updated 3 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆135Updated 3 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆272Updated last year
- ☆96Updated 10 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆528Updated last month
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆108Updated last year
- Materials for learning SGLang☆481Updated last week
- Built upon Megatron-Deepspeed and HuggingFace Trainer, EasyLLM has reorganized the code logic with a focus on usability. While enhancing …☆47Updated 10 months ago
- ☆139Updated last year
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆395Updated 2 weeks ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆212Updated 10 months ago
- A low-latency & high-throughput serving engine for LLMs☆390Updated last month
- PyTorch bindings for CUTLASS grouped GEMM.☆101Updated last month
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆305Updated last week
- GLake: optimizing GPU memory management and IO transmission.☆470Updated 3 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- UltraScale Playbook 中文版☆45Updated 4 months ago
- ⚡️FFPA: Extend FlashAttention-2 with Split-D, achieve ~O(1) SRAM complexity for large headdim, 1.8x~3x↑ vs SDPA.🎉☆192Updated 2 months ago
- ☆53Updated 2 weeks ago