Effective Minimally-Invasive GPU Acceleration of Distributed Sparse Matrix Factorization. - Anshul Gupta; Natalia Gimelshein; Seid Koric; Steven C. Rennich (Euro-Par, 2016)

vDNN: Virtualized deep neural networks for scalable, memory-efficient neural network design. - Minsoo Rhu; Natalia Gimelshein; Jason Clemons; Arslan Zulfiqar; Stephen W. Keckler (MICRO, 2016)

Virtualizing Deep Neural Networks for Memory-Efficient Neural Network Design. - Minsoo Rhu; Natalia Gimelshein; Jason Clemons; Arslan Zulfiqar; Stephen W. Keckler (CoRR, 2016)

Online normalizer calculation for softmax. - Maxim Milakov; Natalia Gimelshein (CoRR, 2018)

PyTorch: An Imperative Style, High-Performance Deep Learning Library. - Adam Paszke; Sam Gross; Francisco Massa; Adam Lerer; James Bradbury; Gregory Chanan; Trevor Killeen; Zeming Lin; Natalia Gimelshein; Luca Antiga; Alban Desmaison; Andreas Köpf; et al. (NeurIPS, 2019)

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)