AIAny
AI Infra2023
Icon for item

torchtitan

Provides a PyTorch-native platform for experimenting with and scaling generative AI training, including composable parallelism, checkpointing, float8, logging, and Llama recipes.

Introduction

Large-model training stacks can become piles of one-off scripts, parallelism libraries, and hardware fixes. torchtitan makes the PyTorch-native path explicit as a compact reference system.

What Sets It Apart

FSDP2, tensor parallelism, pipeline parallelism, context parallelism, checkpointing, torch.compile, float8, and observability are presented as pieces of one training system. Llama recipes make the abstractions concrete.

Who Should Use It

Great fit if you build or evaluate generative AI pretraining infrastructure on PyTorch. Look elsewhere for turnkey training with minimal tuning, PyTorch-independent stacks, or small fine-tuning tools.

Information

  • Websitegithub.com
  • AuthorsPyTorch
  • Published date2023/12/13

More Items

GitHub
AI Infra2025

Measures generative AI inference performance with token-level metrics (TTFT, inter-token latency), latency, and throughput under realistic traffic patterns. Provides a multiprocess engine, real-time TUI dashboard, extensible plugins, and integrations for telemetry and result uploads, aimed at inference benchmarking and capacity planning.

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub

Indexes full text of visited web pages and local files on a self‑hosted server so you can search your personal knowledge from a web UI, terminal, CLI, or an AI assistant. Runs without mandatory telemetry, offers a browser extension for automatic capture, and supports optional semantic search via a configurable embeddings endpoint.