intelligent-machine-learning / atorchLinks
An industrial extension library of pytorch to accelerate large scale model training
☆49Updated last month
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☆240Updated last month
- FlagScale is a large model toolkit based on open-sourced projects.☆354Updated last week
- ☆147Updated 6 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆279Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆151Updated last month
- Zero Bubble Pipeline Parallelism☆427Updated 4 months ago
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆404Updated this week
- ☆95Updated 6 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆139Updated last month
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆245Updated 2 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆106Updated 5 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆570Updated last week
- ☆128Updated 9 months ago
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆408Updated last month
- Training library for Megatron-based models☆85Updated this week
- ☆50Updated this week
- UltraScale Playbook 中文版☆76Updated 6 months ago
- Distributed IO-aware Attention algorithm☆21Updated 3 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆121Updated 3 months ago
- Pipeline Parallelism Emulation and Visualization☆66Updated 3 months ago
- Materials for learning SGLang☆590Updated 3 weeks ago
- ☆88Updated this week
- 青稞Talk☆146Updated last week
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆218Updated last month
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- ☆78Updated 5 months ago
- ☆42Updated last year
- Allow torch tensor memory to be released and resumed later☆135Updated 2 weeks ago
- ☆295Updated this week
- ☆61Updated 2 weeks ago