Pytorch 2: Faster machine learning through dynamic python bytecode transformation and graph compilation

J Ansel, E Yang, H He, N Gimelshein, A Jain… - Proceedings of the 29th …, 2024 - dl.acm.org
This paper introduces two extensions to the popular PyTorch machine learning framework,
TorchDynamo and TorchInductor, which implement the torch. compile feature released in …

TorchTitan: One-stop PyTorch native solution for production ready LLM pre-training

W Liang, T Liu, L Wright, W Constable, A Gu… - arXiv preprint arXiv …, 2024 - arxiv.org
The development of large language models (LLMs) has been instrumental in advancing
state-of-the-art natural language processing applications. Training LLMs with billions of …