5.7k citations · 5.9k across the 11 of their papers we have counts for
Showing 2019Show all
2 papers · 1 filter
cs.CV2019★ 13 cited
Discrete Residual Flow for Probabilistic Pedestrian Behavior Prediction
Ajay Jain, Sergio Casas, Renjie Liao +4
Self-driving vehicles plan around both static and dynamic objects, applying predictive models of behavior to estimate future locations of the objects in the environment. However, f…
cs.LG2019★ 37 cited
Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerialization
Paras Jain, Ajay Jain, Aniruddha Nrusimha +5
We formalize the problem of trading-off DNN training time and memory requirements as the tensor rematerialization optimization problem, a generalization of prior checkpointing stra…