PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays. - View it on GitHub