scan – Looping in PyTensor — PyTensor dev documentation
pytensor.readthedocs.io