Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024 - View it on GitHub