pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
## Description This PR updates the perform method in pytensor/tensor/random/op.py to gracefully handle unimplemented Numba RandomVariables by falling back to object mode. The changes include: Fallback Mechanism: When a RandomVariable...
## Description Wraps [`spicy.signal.gauss_spline`](https://docs.scipy.org/doc/scipy/reference/generated/scipy.signal.gauss_spline.html) as a pytensor `Op` ## Related Issue - [ ] Closes # - [x] Related to https://github.com/pymc-devs/pytensor-workshop/issues/22 ## Checklist - [x] Checked that [the pre-commit linting/style...
Closes #1262 ---- 📚 Documentation preview 📚: https://pytensor--1266.org.readthedocs.build/en/1266/
### Description ```python import pytensor import pytensor.tensor as pt import numpy as np from pytensor.compile.mode import get_mode from pytensor.tensor.blas_c import CGemv A = pt.tensor("A", shape=(None, None)) x = pt.tensor("x", shape=(None,))...
### Description JAX jitting can be insanely slow when there are large constants in the graph. We could add a helper to convert any large constants to symbolic inputs (we...
### Description For Ops with a gufunc_signature, we can automate `infer_shape` implementation: https://github.com/pymc-devs/pytensor/blob/e036caf934e177c72053a40ffbac258c9c26d5ac/pytensor/tensor/slinalg.py#L28-L53 We actually already do it for the Blockwise Wrapper: https://github.com/pymc-devs/pytensor/blob/3cdcfde4155b5aa8d4ac29fd7f07fe21f776a712/pytensor/tensor/blockwise.py#L208-L210
### Description I thought this would be straightforward but we may need something like https://sybil.readthedocs.io/en/latest/quickstart.html My default use of `pytest --doctest-modules --doctest-glob='.*rst'` only works if I specify the docs files...
## Description ## Related Issue - [ ] Closes # - [ ] Related to # ## Checklist - [ ] Checked that [the pre-commit linting/style checks pass](https://docs.pymc.io/en/latest/contributing/python_style.html) - [...
### Description Hit me today, we should use the numpy name
### Description Right now DimShuffle uses an external C-op that is generic over variations of DimShuffle. In the older time it used to be one of the regular unreadable inplice...