pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
## Description All scipy.linalg functions offer an `overwrite_a` and/or `overwrite_b` argument that can enhance performance by re-using the input memory for the outputs. This PR implements re-writes that will set...
### Description If a user wants to operate with the non-default output of RVs, they have to do `rv.make_node()` or `rv().owner.outputs` none of which is ideal or very readable. I...
### Description RandomVariables third input is a numerical integer than indexes into a list of predefined types to indicate if a variable is float64 or float32, ... This input is...
### Description If we only have on instance of the same OpFromGraph Op we should move constant inputs inside, to allow a simpler internal graph at compilation time. We shoud...
## Description I think I got all the unittest_tools.assert_allclose instances The following files no longer used functions from unittest_tools so I removed the import tests/link/test_vm.py tests/tensor/test_variable.py tests/scalar/test_basic.py tests/sparse/test_rewriting.py tests/link/c/test_params_type.py tests/tensor/rewriting/test_elemwise.py...
Numpy 2.0
## Description Ruff suggested a few things to change but I cannot test because I install both `numpy=2.0.0.rc1` and `numba` (any version including the most recent `0.59.1`) (incompatibilities in python...
### Description As discussed in https://github.com/pymc-devs/pymc/pull/7212#discussion_r1553967791 it's not clear what the signature / type is. Besides a Mode object there is a set of strings that are allowed: https://github.com/pymc-devs/pytensor/blob/a76172ee6a9753266150fc7bac4dd66906967a26/pytensor/compile/mode.py#L484-L534
Works: ```python import pytensor import pytensor.tensor as at import numpy as np data = at.constant(np.random.randn(64)) srng = pytensor.tensor.random.RandomStream() index = srng.integers(64, size=10) datai = data[index] var = at.vector("var") scan =...
### Description Now that #306 is merged, there are a couple of follow ups we should do: 1. Blockwise more Ops (Everything in linalg?) 2. Dispatch vectorize for more Ops...
### Description [This paper ](https://arxiv.org/abs/2309.03060) and [this library ](https://github.com/wilson-labs/cola)describe and implement a number of linear algebra simplifications that we can implement as graph rewrites. This issue is both a tracker...