Stochastic Taylor Derivative Estimator: Efficient amortization for arbitrary differential operators

Neural Information Processing Systems 

Our code is available at https://github.com/sail-sg/stde