Minimal PyTorch implementation of TP, SP, FSDP and sharded-EMA
☆32Nov 27, 2025Updated 11 months ago
Alternatives and similar repositories for FSDP-Training
Users that are interested in FSDP-Training are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Resource Multiplexing in Tuning and Serving Large Language Models (USENIX ATC 2025)☆10Apr 21, 2026Updated 6 months ago
- A living arena for Human vs Agent and Agent vs Agent competition, testing OpenClaw intelligence through real-time gameplay.☆13Mar 18, 2026Updated 7 months ago
- OSP-Next☆68Jun 22, 2026Updated 4 months ago
- SC'25 UltraAttn: Efficiently Parallelizing Attention through Hierarchical Context-Tiling☆16Aug 14, 2025Updated last year
- ☆10Oct 22, 2026Updated last week
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆11May 21, 2026Updated 5 months ago
- Mini-SGLang is a mini version of SGLang project ().☆13Sep 24, 2025Updated last year
- A scheduling framework for multitasking over diverse XPUs, including GPUs, NPUs, ASICs, and FPGAs☆14Jun 17, 2026Updated 4 months ago
- A lightweight CuTe-based CUDA kernel for DuoAttention, optimized for large language model inference.☆12Dec 25, 2025Updated 10 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year
- ☆40Dec 14, 2025Updated 10 months ago
- Code repository for the SOSP'25 paper DCP: Addressing Input Dynamism In Long-Context Training via Dynamic Context Parallelism.☆23Nov 28, 2025Updated 11 months ago
- Spark-Wan: Few-Step Video Diffusion Transformer via Hybrid Adversarial Post-Training☆18Mar 8, 2026Updated 7 months ago
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Sep 11, 2026Updated last month
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆54May 19, 2025Updated last year
- ☆46Oct 15, 2025Updated last year
- Tiny-FSDP, a minimalistic re-implementation of the PyTorch FSDP☆111Aug 20, 2025Updated last year
- MetaAttention: A Unified and Performant Attention Framework Across Hardware Backends(PPoPP'26)☆18Aug 6, 2026Updated 2 months ago
- ☆23May 5, 2025Updated last year
- Distillation pipeline from pretrained Transformers to customized FLA models☆13Aug 4, 2025Updated last year
- Some funny cute/cuteDSL code snippets☆35Mar 2, 2026Updated 7 months ago
- Official code for ReBA：Better load balancing for VLM-MoE☆20Aug 4, 2026Updated 2 months ago
- ☆12Nov 21, 2025Updated 11 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Multi-Level Triton Runner supporting Python, IR, PTX, AMDGCN, cubin and hasco.☆100Sep 17, 2026Updated last month
- A Triton-only attention backend for vLLM☆28Jul 14, 2026Updated 3 months ago
- LLMBind: A Unified Modality-Task Integration Framework☆19Jun 16, 2024Updated 2 years ago
- ☆32Jul 2, 2025Updated last year
- [MLSys 26] 🥇 Solution for Gated Delta Net Track of MLSys 26 Flash infer competition☆36May 22, 2026Updated 5 months ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆53Aug 20, 2025Updated last year
- NCU-driven iterative optimization workflow for CUDA/CUTLASS/Triton/CuTe DSL kernels.☆25Apr 10, 2026Updated 6 months ago
- iFSQ & LlamaGen-REPA☆107Jan 27, 2026Updated 9 months ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 9 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Offical implementation of "Spike2Former: Efficient Spiking Transformer for High-performance Image Segmentation" (AAAI2025 Oral)☆45Oct 10, 2026Updated 2 weeks ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- BlitzScale Router - Distributed LLM Inference Router (Rust)☆17May 25, 2026Updated 5 months ago
- [EuroSys27] TokenCake: A KV-Cache-centric Serving Framework for LLM-based Multi-Agent Applications☆14Oct 14, 2026Updated 2 weeks ago
- a simple API to use CUPTI☆10Aug 19, 2025Updated last year
- [AAAI26] Next Patch Prediction☆129Jan 2, 2025Updated last year