Best practices & guides on how to write distributed pytorch training code
☆581Oct 22, 2025Updated 4 months ago
Alternatives and similar repositories for distributed-training-guide
Users that are interested in distributed-training-guide are comparing it to the libraries listed below
Sorting:
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,752Jul 18, 2025Updated 7 months ago
- ☆239Nov 24, 2025Updated 3 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆133Dec 3, 2024Updated last year
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,090Aug 26, 2025Updated 6 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆280Nov 24, 2025Updated 3 months ago
- Efficient Triton Kernels for LLM Training☆6,162Updated this week
- Supercharge huggingface transformers with model parallelism.☆78Jul 23, 2025Updated 7 months ago
- For optimization algorithm research and development.☆558Feb 23, 2026Updated last week
- Exploring Applications of GRPO☆251Aug 25, 2025Updated 6 months ago
- What would you do with 1000 H100s...☆1,154Jan 10, 2024Updated 2 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆595Aug 12, 2025Updated 6 months ago
- ☆15Mar 2, 2025Updated last year
- Source code of our paper "PairDistill: Pairwise Relevance Distillation for Dense Retrieval", EMNLP 2024 Main.☆22Nov 28, 2024Updated last year
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 9 months ago
- A lightweight, low-dependency, unified API to use all common reranking and cross-encoder models.☆1,599Dec 20, 2025Updated 2 months ago
- NanoGPT (124M) in 2 minutes☆4,679Updated this week
- Minimalistic large language model 3D-parallelism training☆2,579Feb 19, 2026Updated last week
- Machine Learning Engineering Open Book☆17,162Feb 21, 2026Updated last week
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- AnchorAttention: Improved attention for LLMs long-context training☆214Jan 15, 2025Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆201Jun 1, 2025Updated 9 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,428Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,707Updated this week
- Initialization using Update Approximation is a Silver Bullet for Extremely Efficient Low-Rank Fine-Tuning☆52Oct 17, 2025Updated 4 months ago
- PyTorch native post-training library☆5,691Updated this week
- code for training & evaluating Contextual Document Embedding models☆201May 14, 2025Updated 9 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,184Aug 22, 2025Updated 6 months ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆597Oct 7, 2025Updated 4 months ago
- ☆48Aug 29, 2024Updated last year
- (WIP) A small but powerful, homemade PyTorch from scratch.☆676Feb 24, 2026Updated last week
- Transformers-compatible library for applying various compression algorithms to LLMs for optimized deployment with vLLM☆2,787Updated this week
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆59Oct 18, 2025Updated 4 months ago
- XVERSE-MoE-A36B: A multilingual large language model developed by XVERSE Technology Inc.☆39Sep 12, 2024Updated last year
- YaFSDP: Yet another Fully Sharded Data Parallel☆985Jan 30, 2026Updated last month
- ☆33Aug 9, 2024Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,550Jan 12, 2025Updated last year