Allowed samplers to store gradients (HMC and NUTS are pending on that respect)