torchtune: PyTorch native post-training library

Mark Obozov, Maxime Griot, Joseph Cummings, Evan Smothers, Felipe Mello +6 more
5/20/2026
cs.LGcs.AI

No official code found. Best implementation: meta-pytorch/torchtune (5,812 stars, BSD-3-Clause, last commit 2026-05-20). 1 more below.

Abstract

Modern LLMs typically require multistage training pipelines to achieve strong downstream performance, with post-training serving as the main interface for adapting open-weight models. We introduce torchtune, a PyTorch-native library designed to streamline the post-training lifecycle of LLMs, enabling efficient fine-tuning, experimentation, and deployment-oriented workflows. Unlike many existing fine-tuning frameworks, which often optimize for ease of use, specialized recipes, or hardware efficiency at the cost of transparency and extensibility, torchtune emphasizes modularity, hackability, and direct access to the underlying PyTorch components. In this paper, we present the design principles behind torchtune, describe how they are reflected in its model builders, training recipes, and distributed training stack, and evaluate the library across representative post-training settings. We compare against popular fine-tuning frameworks, including Axolotl and Unsloth, and show that torchtune provides strong performance and memory efficiency across many settings while remaining flexible enough for rapid research iteration. These results position torchtune as a practical foundation for reproducible LLMs post-training research.

View on arXivView PDF

Code Implementations(2)

BSD-3-Clause

PyTorch native post-training library

5,812757Oct 20, 20235 months agoBSD-3-Clause
BSD-3-Clause

PyTorch native post-training library

00Feb 14, 20251 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{obozov2026torchtune,
  title  = {torchtune: PyTorch native post-training library},
  author = {Mark Obozov and Maxime Griot and Joseph Cummings and Evan Smothers and Felipe Mello and Rafi Ayub and Philip John Bontrager and Salman Mohammadi and Ariel Kwiatkowski and Nathan Azrak and Mircea Mironenco},
  year   = {2026},
  eprint = {2605.21442},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2605.21442v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2605.21442.svg)](https://codeofpaper.com/paper/2605.21442)

Discussion