torchtitan AI Agent Skills

distributed-llm-pretraining-torchtitan

21.8k

davila7

Provides PyTorch-native distributed LLM pretraining using torchtitan with 4D parallelism (FSDP2, TP, PP, CP). Use when pretraining Llama 3.1, DeepSeek V3, or custom models at scale from 8 to 512+ GPUs with Float8, torch.compile, and distributed checkpointing.

torchforge-rl-training

21.8k

davila7

Provides guidance for PyTorch-native agentic RL using torchforge, Meta's library separating infrastructure from algorithms. Use when you want clean RL abstractions, easy algorithm experimentation, or scalable training with Monarch and TorchTitan.