A PyTorch native platform for training generative AI models
☆5,794Oct 10, 2026Updated this week
Alternatives and similar repositories for torchtitan
Users that are interested in torchtitan are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyTorch native post-training library☆5,811Sep 9, 2026Updated last month
- Efficient Triton Kernels for LLM Training☆6,653Updated this week
- PyTorch native quantization for training and inference☆2,997Updated this week
- Minimalistic large language model 3D-parallelism training☆2,835Updated this week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…☆3,572Updated this week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Tile primitives for speedy kernels☆3,746Sep 12, 2026Updated 3 weeks ago
- Ongoing research training transformer models at scale☆18,108Updated this week
- 🚀 Efficient implementations for emerging model architectures☆5,842Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,321Aug 26, 2025Updated last year
- Fast and memory-efficient exact attention☆25,115Updated this week
- Development repository for the Triton language and compiler☆20,337Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,258Aug 22, 2025Updated last year
- FlashInfer: Kernel Library for LLM Serving☆6,584Updated this week
- Helpful tools and examples for working with flex-attention☆1,257Updated this week
- 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.
- Ring attention implementation with flash attention☆1,064Sep 10, 2025Updated last year
- Distributed Compiler and Optimized Parallel Kernels☆1,561Sep 18, 2026Updated 3 weeks ago
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,769Jul 18, 2025Updated last year
- verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework☆23,822Updated this week
- ☆1,595Mar 25, 2026Updated 6 months ago
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,370Aug 28, 2025Updated last year
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,561Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆960Updated this week
- NanoGPT (124M) in 90 seconds☆5,931Sep 28, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆543Updated this week
- SGLang is a high-performance serving framework for large language models and multimodal models.☆36,962Updated this week
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated 2 years ago
- VeOmni: Scaling Any Modality Model Training with Model-Centric Distributed Recipe Zoo☆2,234Updated this week
- slime is an LLM post-training framework for RL Scaling.☆8,626Updated this week
- A Quirky Assortment of CuTe Kernels☆1,167Sep 17, 2026Updated 3 weeks ago
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆10,554Sep 23, 2026Updated 2 weeks ago
- TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizat…☆14,787Updated this week
- Train transformer language models with reinforcement learning.☆19,483Updated this week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,911Updated this week
- Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs☆1,041Mar 3, 2026Updated 7 months ago
- PyTorch extensions for high performance and large scale training.☆3,404Apr 26, 2025Updated last year
- Puzzles for learning Triton☆2,628Apr 1, 2026Updated 6 months ago
- Accessible large language models via k-bit quantization for PyTorch.☆8,517Sep 7, 2026Updated last month
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆8,557Updated this week
- Modeling, training, eval, and inference code for OLMo☆6,697Nov 24, 2025Updated 10 months ago