Unlock the Power of THOAD: Revolutionize PyTorch Graphs with High-Order Derivatives
Intro
I’m excited to share thoad (short for PyTorch High Order Automatic Differentiation), a Python only library that computes arbitrary order partial derivatives directly on a PyTorch comp...