Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch - View it on GitHub