pytensor

pytensor copied to clipboard

Add Ops for LU Factorization

2

## Description This PR will add the following Ops: - [x] lu - [x] lu_factor - [x] lu_solve As well as dispatches for numba/jax (and maybe torch, though help is...

jessegrabowski

Simplify Numba implementation of Alloc

1

Spinoff from #811 ---- 📚 Documentation preview 📚: https://pytensor--1329.org.readthedocs.build/en/1329/

ricardoV94

## 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) - [...

ricardoV94

Use `shell=False` for `Popen` on Windows

3

## Description I'm having hard times creating new environments with working compiler dependencies. My debugging attempts show that `shell=False` could fix this, however this observation contradicts the comments next to...

michaelosthege

Add Ops for specialized dot products with structured matrices

3

### Description BLAS/LAPACK offer a number of specialized dot products when matrices have special structure: - Triangular [matrix-vector product ](https://netlib.org/lapack/explore-html/d6/d1c/group__trmv_ga73370bd6dca01abe05d54ecd1d91ce9a.html#ga73370bd6dca01abe05d54ecd1d91ce9a) and [matrix-matrix product](https://netlib.org/lapack/explore-html/dd/dab/group__trmm_ga4d2f76d6726f53c69031a2fe7f999add.html#ga4d2f76d6726f53c69031a2fe7f999add) - Symmetrical [matrix-vector product ](https://netlib.org/lapack/explore-html/d0/d4b/group__hpmv_ga739f8dc2316523832bde2b237fcad8a6.html#ga739f8dc2316523832bde2b237fcad8a6) and [matrix-matrix...

jessegrabowski

Inplace Composite and ScalarLoop Ops with multiple outputs

2

Closes #138 This allows inplacing arbitrary number of inputs in Elemwise Composite (i.e., fused) Ops. The restriction was there because in the original codegen writing to the aliased outputs could...

ricardoV94

### Description After we get #1318 we can offer 1d correlate trivially. It's just convolve with the second input transposed (and perhaps conjugated if complex)

ricardoV94

This PR simplifies the expressions returned by `Subtensor.infer_shape`. Some examples: This PR also implements some basic minmax algebra rewrites to handle constraints implied by slice inputs. The following shape graph...

ricardoV94

[pre-commit.ci] pre-commit autoupdate

2

updates: - [github.com/astral-sh/ruff-pre-commit: v0.14.0 → v0.14.2](https://github.com/astral-sh/ruff-pre-commit/compare/v0.14.0...v0.14.2) ---- 📚 Documentation preview 📚: https://pytensor--1095.org.readthedocs.build/en/1095/

pre-commit-ci[bot]

### Description Follow up to #1318 Some of these may apply to higher-dimensional convolutions. ## Sum of convolution as product of sum of inputs If we have `convolve1d(x, y, mode="full").sum()`,...

ricardoV94