pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
### Description If I replace a tensor with shape `(None,)` by one with static shape `(1,)` llvm aborts during codegen for me. Maybe some rewrite doesn't behave properly, and we...
## Description ### TODO: - [x] Merge code duplication into elemwise_codegen - [x] Handle broadcast via size argument - [x] Handle more than normal - [x] Add expand_dims in make_node....
## Description This showed up in #691 and was introduced/not properly handled by #664 We need to explicitly broadcast the new parameters to define the new size! ## Checklist -...
## Description Makes graphs with Assert a bit more readable ## Checklist - [x] Checked that [the pre-commit linting/style checks pass](https://docs.pymc.io/en/latest/contributing/python_style.html) - [x] Included tests that prove the fix is...
### Description #735 shows how the current interface to copy stack traces during rewrites look like. It's verbose and easy to forget. There is/was a special mode/flag to check variables...
### Description https://github.com/pymc-devs/pytensor/actions/runs/8894500236/job/24423790167?pr=735#step:6:40
## Description We should keep the stack trace during our rewrites, so that when the Op fails to perform we can point to the moment where the node was created....
### Description SVD comes with a bunch of keyword arguments, most important of which is `compute_uv`. If False, it will return only the singular values for a given matrix. This...
### Description As discussed in [#693](https://github.com/pymc-devs/pytensor/pull/693#discussion_r1562285879), we'd like to avoid unnecessary print statements before committing the changes itself. This way, we can avoid unnecessary debug statements. PyMC already has this...
## Description Reverting back to the option first suggested in https://github.com/pymc-devs/pytensor/pull/493 and then overridden in https://github.com/pymc-devs/pytensor/pull/494 The `[ ]` is the only way that allows for the Python slice `:`...