Using Python for Model Inference in Deep Learning. - Zachary DeVito; Jason Ansel; Will Constable; Michael Suo; Ailing Zhang; Kim M. Hazelwood (CoRR, 2021)

TorchBench: Benchmarking PyTorch with High API Surface Coverage. - Yueming Hao; Xu Zhao 004; Bin Bao; David Berard; Will Constable; Adnan Aziz; Xu Liu (CoRR, 2023)

PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation. - Jason Ansel; Edward Z. Yang; Horace He; Natalia Gimelshein; Animesh Jain; Michael Voznesensky; Bin Bao; Peter Bell 008; David Berard; Evgeni Burovski; Geeta Chauhan; Anjali Chourdia; et al. (ASPLOS (2), 2024)

TorchTitan: One-stop PyTorch native solution for production ready LLM pre-training. - Wanchao Liang; Tianyu Liu; Less Wright; Will Constable; Andrew Gu; Chien-Chin Huang; Iris Zhang; Wei Feng; Howard Huang; Junjie Wang; Sanket Purandare; Gokul Nadathur; et al. (CoRR, 2024)

TorchTitan: One-stop PyTorch native solution for production ready LLM pretraining. - Wanchao Liang; Tianyu Liu; Less Wright; Will Constable; Andrew Gu; Chien-Chin Huang; Iris Zhang; Wei Feng; Howard Huang; Junjie Wang; Sanket Purandare; Gokul Nadathur; et al. (ICLR, 2025)