Tiny-FSDP, a minimalistic re-implementation of the PyTorch FSDP
☆99Aug 20, 2025Updated 6 months ago
Alternatives and similar repositories for Tiny-FSDP
Users that are interested in Tiny-FSDP are comparing it to the libraries listed below
Sorting:
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆50Aug 20, 2025Updated 6 months ago
- Asynchronous pipeline parallel optimization☆19Feb 2, 2026Updated last month
- Tiny-Megatron, a minimalistic re-implementation of the Megatron library☆23Sep 1, 2025Updated 6 months ago
- ☆126Aug 10, 2025Updated 6 months ago
- ☆10Nov 18, 2024Updated last year
- a simple API to use CUPTI☆11Aug 19, 2025Updated 6 months ago
- ☆24May 9, 2025Updated 9 months ago
- [ICML‘25] Official code for paper "Occult: Optimizing Collaborative Communication across Experts for Accelerated Parallel MoE Training an…☆13Apr 17, 2025Updated 10 months ago
- ☆35Apr 8, 2025Updated 10 months ago
- LoRAFusion: Efficient LoRA Fine-Tuning for LLMs☆24Sep 23, 2025Updated 5 months ago
- Implementation from scratch in C of the Multi-head latent attention used in the Deepseek-v3 technical paper.☆18Jan 15, 2025Updated last year
- Official code for the paper "HEXA-MoE: Efficient and Heterogeneous-Aware MoE Acceleration with Zero Computation Redundancy"☆15Mar 6, 2025Updated last year
- Inferix: A Block-Diffusion based Next-Generation Inference Engine for World Simulation☆110Feb 27, 2026Updated last week
- AI model training on heterogeneous, geo-distributed resources☆38Nov 24, 2025Updated 3 months ago
- MiroTrain is an efficient and algorithm-first framework research agent.☆133Aug 27, 2025Updated 6 months ago
- 模仿 TensorFlow 写的极简深度学习框架，仅供练习目的☆14Sep 11, 2018Updated 7 years ago
- A lightweight Inference Engine built for block diffusion models☆41Dec 9, 2025Updated 2 months ago
- APEX+ is an LLM Serving Simulator☆42Jun 16, 2025Updated 8 months ago
- LLMBind: A Unified Modality-Task Integration Framework☆19Jun 16, 2024Updated last year
- Allow torch tensor memory to be released and resumed later☆220Feb 9, 2026Updated 3 weeks ago
- ☆20Dec 2, 2024Updated last year
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago
- ☆32Jan 4, 2026Updated 2 months ago
- PoC for "SpecReason: Fast and Accurate Inference-Time Compute via Speculative Reasoning" [NeurIPS '25]☆65Oct 2, 2025Updated 5 months ago
- Compression for Foundation Models☆35Jul 21, 2025Updated 7 months ago
- ☆38Mar 24, 2012Updated 13 years ago
- The official implementation for the intra-stage fusion technique introduced in https://arxiv.org/abs/2409.13221☆31Apr 22, 2025Updated 10 months ago
- [MM 2025] Towards Modality Generalization: A Benchmark and Prospective Analysis☆28May 22, 2025Updated 9 months ago
- Dynamic resources changes for multi-dimensional parallelism training☆30Aug 22, 2025Updated 6 months ago
- Minimal RLHF implementation built on top of minGPT.☆32Jul 4, 2024Updated last year
- (WIP) Parallel inference for black-forest-labs' FLUX model.☆19Nov 18, 2024Updated last year
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆101Aug 25, 2025Updated 6 months ago
- ☆131Nov 11, 2024Updated last year
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆127Nov 10, 2025Updated 3 months ago
- ☆27May 27, 2024Updated last year
- This repository provides an improved LLamaGen Model, fine-tuned on 500,000 high-quality images, each accompanied by over 300 token prompt…☆30Oct 21, 2024Updated last year
- LLM training technologies developed by kwai☆70Jan 21, 2026Updated last month
- Official code for the paper: Can3Tok (ICCV2025)☆39Aug 23, 2025Updated 6 months ago
- 用python 实现一个简单的深度学习框架☆31Apr 3, 2021Updated 4 years ago