intelligent-machine-learning / atorchLinks
An industrial extension library of pytorch to accelerate large scale model training
☆58Updated 5 months ago
Alternatives and similar repositories for atorch
Users that are interested in atorch are comparing it to the libraries listed below
Sorting:
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆258Updated 6 months ago
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆269Updated last week
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆47Updated last year
- ☆449Updated 5 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆287Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆79Updated last year
- ☆131Updated 8 months ago
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆55Updated 3 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆154Updated 5 months ago
- ☆155Updated 11 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆641Updated 3 weeks ago
- ☆118Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆184Updated last month
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆99Updated 5 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆250Updated this week
- mllm-npu: training multimodal large language models on Ascend NPUs☆95Updated last year
- Zero Bubble Pipeline Parallelism☆449Updated 9 months ago
- 青稞Talk☆190Updated 2 weeks ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆190Updated last week
- Model compression toolkit engineered for enhanced usability, comprehensiveness, and efficiency.☆314Updated this week
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆110Updated 10 months ago
- Distributed IO-aware Attention algorithm☆24Updated 4 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆474Updated this week
- ☆87Updated last week
- High performance inference engine for diffusion models☆103Updated 5 months ago
- ☆130Updated 5 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆370Updated 6 months ago
- Efficient triton implementation of Native Sparse Attention.☆262Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆142Updated 8 months ago