pytensor

pytensor copied to clipboard

Use vectorized jacobian in minimize ops

5

This PR uses the more new experimental vectorized jacobian in the minimize/root Ops, with the goal of getting it more tested, as it won't be breaking any people's code. The...

ricardoV94

### Description ```python from pytensor.graph import FunctionGraph import pytensor.scalar as ps from pytensor.link.c.basic import CLinker x = ps.float64("x") o = x + 1 cl = CLinker().accept(FunctionGraph([x], [o])) cl.cmodule_key() from pytensor.graph...

ricardoV94

Second gradient of simple Scan shows some missing simplifications

3

### Description ```python import pytensor import pytensor.tensor as pt from pytensor.compile.mode import Mode x0 = pt.scalar("x0") ys, _ = pytensor.scan( lambda ytm1: ytm1 ** 2, outputs_info=[x0], n_steps=4, mode=Mode(linker="py", optimizer="fast_run").excluding("fusion"), )...

ricardoV94

Add rewrite to fuse nested `BlockDiag` Ops

2

### Description This comes up in statespace models. `BlockDiag` is variadic, so graphs like this can be canonicalized: ```py import pytensor.tensor as pt import pytensor a, b, c = pt.matrices('a',...

jessegrabowski

Rename `pytensor.tensor.math.sub` to `pytensor.tensor.math.subtract` to match array API

1

https://github.com/pymc-devs/pytensor/blob/82bb9642daa7514c07e233b3486bd75f85503d8c/pytensor/sparse/basic.py?plain=1#L2273

williambdean

Rename `pytensor.tensor.math.mul` to `pytensor.tensor.math.multiply` to match array API standard

1

https://github.com/pymc-devs/pytensor/blob/82bb9642daa7514c07e233b3486bd75f85503d8c/pytensor/sparse/basic.py?plain=1#L2496

williambdean

Rewrite nested incsubtensor

1

## 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

### Issue with current documentation: We are missing examples using MLX backend. ### Idea or request for content: _No response_

cetagostini

Extend verify_grad to complex gradient

4

## Description Extend verify_grad to complex gradient following the holomorphic gradient convention (as in JAX). The decision on which convention to follow (JAX-like or torch-like) has not been taken yet;...

educhesne

### Description Why would function be inside `types.py`? Better name suggestions? This is utmost priority!

ricardoV94