Faster SGD training by minibatch persistency
Automatic differentiation in PyTorch