HMC and NUTS store gradients now