Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Picture for Natalia Gimelshein

Natalia Gimelshein

NVIDIA

PyTorch: An Imperative Style, High-Performance Deep Learning Library


Dec 03, 2019
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, Edward Yang, Zach DeVito, Martin Raison, Alykhan Tejani, Sasank Chilamkurthy, Benoit Steiner, Lu Fang, Junjie Bai, Soumith Chintala

* 12 pages, 3 figures, NeurIPS 2019 

  Access Paper or Ask Questions

Online normalizer calculation for softmax


Jul 28, 2018
Maxim Milakov, Natalia Gimelshein

* 1) Added link to the benchmark code, 2) Benchmarked Safe Softmax + Top-K fused and attributed part of 5x explicitly to fusion in sections 5.2 and 6, 3) Stylistic changes, 4) Minor clarifications 

  Access Paper or Ask Questions

vDNN: Virtualized Deep Neural Networks for Scalable, Memory-Efficient Neural Network Design


Jul 28, 2016
Minsoo Rhu, Natalia Gimelshein, Jason Clemons, Arslan Zulfiqar, Stephen W. Keckler

* Published as a conference paper at the 49th IEEE/ACM International Symposium on Microarchitecture (MICRO-49), 2016 

  Access Paper or Ask Questions