Description:

PyTensor is a Python library that allows you to define, optimize, and
evaluate mathematical expressions involving multi-dimensional arrays. PyTensor
features:
* A hackable, pure-Python codebase
* Extensible graph framework suitable for rapid development of custom operators
and symbolic optimizations
* Implements an extensible graph transpilation framework that currently provides
compilation via C, JAX, and Numba
* Contrary to PyTorch and TensorFlow, PyTensor maintains a static graph which
can be modified in-place to allow for advanced optimizations