PyTorch Automatic Differentiation Forward Mode and Reverse Mode Using autograd and functorch - View it on GitHub