NumTorch is a Python library that offers tensor computation through the NumPy library and an automatic differentiation system for building dynamic computation graphs.

Created as a learning project from Trask, in Grokking Deep Learning (2019), its API is similar to that of PyTorch.

Like many of my projects, this has primarily been built for my use cases. If you wish to extend the base functionality, you're encouraged to fork the project.