intelligent-machine-learning / atorchLinks
An industrial extension library of pytorch to accelerate large scale model training
☆52Updated 3 months ago
Alternatives and similar repositories for atorch
Users that are interested in atorch are comparing it to the libraries listed below
Sorting:
- A collection of memory efficient attention operators implemented in the Triton language.☆283Updated last year
- FlagScale is a large model toolkit based on open-sourced projects.☆412Updated last week
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆250Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆169Updated last month
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- Training library for Megatron-based models☆209Updated last week
- ☆152Updated 8 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆146Updated 3 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆602Updated last month
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆252Updated 4 months ago
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆498Updated this week
- ☆65Updated 2 weeks ago
- ☆97Updated 8 months ago
- Distributed IO-aware Attention algorithm☆22Updated 2 months ago
- Model compression toolkit engineered for enhanced usability, comprehensiveness, and efficiency.☆207Updated last week
- Odysseus: Playground of LLM Sequence Parallelism☆78Updated last year
- ☆438Updated 3 months ago
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆412Updated 3 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆218Updated last year
- UltraScale Playbook 中文版☆90Updated 8 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆109Updated 7 months ago
- 青稞Talk☆168Updated this week
- Pipeline Parallelism Emulation and Visualization☆71Updated 5 months ago
- A quantization algorithm for LLM☆146Updated last year
- ☆130Updated 11 months ago
- ☆132Updated 6 months ago
- ☆79Updated last year
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆83Updated 3 months ago
- ☆111Updated 6 months ago
- Zero Bubble Pipeline Parallelism☆435Updated 6 months ago