computational-graph
A Pure Python Deep Learning Framework with Automatic Differentiation and PyTorch-like API
Lightweight performat Python 3.12+ automatic differentiation system that leverages PyTorch’s computational graph to compute arbitrary-order partial derivatives.