intelligent-machine-learning / atorchLinks
An industrial extension library of pytorch to accelerate large scale model training
☆49Updated 2 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☆241Updated 2 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆281Updated last year
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆144Updated last month
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- FlagScale is a large model toolkit based on open-sourced projects.☆362Updated this week
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆428Updated this week
- ☆148Updated 7 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆154Updated last week
- Odysseus: Playground of LLM Sequence Parallelism☆78Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆107Updated 6 months ago
- Training library for Megatron-based models☆125Updated this week
- ☆96Updated 6 months ago
- ☆129Updated 9 months ago
- 青稞Talk☆151Updated this week
- Pipeline Parallelism Emulation and Visualization☆67Updated 4 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆223Updated 2 months ago
- Distributed IO-aware Attention algorithm☆21Updated 3 weeks ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- ☆91Updated this week
- ☆120Updated 2 months ago
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆68Updated last month
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆576Updated this week
- Zero Bubble Pipeline Parallelism☆432Updated 5 months ago
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆249Updated 3 months ago
- High performance inference engine for diffusion models☆94Updated last month
- A quantization algorithm for LLM☆143Updated last year
- ☆78Updated 6 months ago
- Built upon Megatron-Deepspeed and HuggingFace Trainer, EasyLLM has reorganized the code logic with a focus on usability. While enhancing …☆48Updated last year
- ☆428Updated 2 months ago
- ☆130Updated 4 months ago