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