activity
20132025
most citedPyTorch: An Imperative Style, High-Performance Deep Learning Library

16.2k citations · 16.4k across the 7 of their papers we have counts for

collaborators

7 papers

cs.CL2025

Balancing Knowledge Delivery and Emotional Comfort in Healthcare Conversational Systems

Shang-Chi Tsai, Yun-Nung Chen

With the advancement of large language models, many dialogue systems are now capable of providing reasonable and informative responses to patients' medical conditions. However, whe…

cs.CL2024★ 164 cited

GPT-4o System Card

OpenAI, :, Aaron Hurst +416

GPT-4o is an autoregressive omni model that accepts as input any combination of text, audio, image, and video, and generates any combination of text, audio, and image outputs. It's…

cs.CV2023★ 2 cited

One-Step Diffusion Distillation via Deep Equilibrium Models

Zhengyang Geng, Ashwini Pokle, J. Zico Kolter

Diffusion models excel at producing high-quality samples but naively require hundreds of iterations, prompting multiple attempts to distill the generation process into a faster net…

cs.LG2019★ 16.2k cited

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

Adam Paszke, Sam Gross, Francisco Massa +18

Deep learning frameworks have often focused on either usability or speed, but not both. PyTorch is a machine learning library that shows that these two goals are in fact compatible…

cs.PF2018★ 20 cited

Online normalizer calculation for softmax

Maxim Milakov, Natalia Gimelshein

The Softmax function is ubiquitous in machine learning, multiple previous works suggested faster alternatives for it. In this paper we propose a way to compute classical Softmax wi…

cs.DC2016★ 1 cited

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

Minsoo Rhu, Natalia Gimelshein, Jason Clemons +2

The most widely used machine learning frameworks require users to carefully tune their memory usage so that the deep neural network (DNN) fits into the DRAM capacity of a GPU. This…

physics.flu-dyn2013

Numerical modeling of ion transport in a ESI-MS system

N. Gimelshein, S. Gimelshein, T. Lilly +1

Gas and ion transport in the capillary-skimmer subatmospheric interface of a mass spectrometer, which is typically utilized to separate unevaporated micro-droplets from ions, was s…