PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation

Ansel, Jason; Yang, Edward; He, Horace; Gimelshein, Natalia; Jain, Animesh; Voznesensky, Michael; Bao, Bin; Bell, Peter; Berard, David; Burovski, Evgeni; Chauhan, Geeta; Chourdia, Anjali; Constable, Will; Desmaison, Alban; DeVito, Zachary; Ellison, Elias; Feng, Will; Gong, Jiong; Gschwind, Michael; Hirsh, Brian; Huang, Sherlock; Kalambarkar, Kshiteej; Kirsch, Laurent; Lazos, Michael; Lezcano, Mario; Liang, Yanbo; Liang, Jason; Lu, Yinghai; Luk, C. K.; Maher, Bert; Pan, Yunjie; Puhrsch, Christian; Reso, Matthias; Saroufim, Mark; Siraichi, Marcos Yukio; Suk, Helen; Zhang, Shunting; Suo, Michael; Tillet, Phil; Zhao, Xu; Wang, Eikan; Zhou, Keren; Zou, Richard; Wang, Xiaodong; Mathews, Ajit; Wen, William; Chanan, Gregory; Wu, Peng; Chintala, Soumith

doi:10.1145/3620665.3640366

Public

PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation

Published in Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2 • Apr 27, 2024

NobleIDNI6P03W71R20S96

Authors:

Jason Ansel

,

Edward Yang

,

Horace He

Finding related papers...

Discussions

(0)

No comments yet

Be the first to share your thoughts!