Jesse Grabowski

JAX backend fails for latent scan variables

14

### Describe the issue: Not sure if this belongs here or in the pytensor repo. Putting it here because the minimal example I can come up with uses PyMC. If...

### Description We currently have a long list of if/else conditions. but it's needlessly restrictive. Basically the user is allowed to select an initialization mode (`jitter`, `MAP`, or `advi`) and...

Creating Distributions with `CustomDist`

3

# Notebook proposal **Title**: Creating Distributions with `CustomDist` ## Why should this notebook be added to pymc-examples? This comes up a lot on the discourse, most recently [here](https://discourse.pymc.io/t/how-to-implement-bivariate-poisson-logbp/16400?u=jessegrabowski). New users...

This PR does two things: 1. I added a minimial docstring to `make_transform_adapt` to give a basic outline of each parameter (as I understand them), and try to explain situations...

## Description This test is flakey and responsible for a lot of random headaches. It does not appear to be required to fail? The test links to issue #770, but...

Handle slices in `mlx_funcify_IncSubtensor`

8

## Description The MLX dispatch for `IncSubtensor` was assuming that the indexes would always be integers, but they can actually be either integers or slices. This PR adds logic to...

Add linalg Ops to MLX backend

3

## Description This PR adds mlx dispatches for the following Ops: - `nlinalg.kron` - `nlinalg.svd` - `nlinalg.inv` - `nlinalg.pinv` - `slinalg.cholesky` - `slinalg.solve` - `slinalg.solve_triangular` - `slinalg.lu` Most (all?) of...

Implement `Convolve2D` Op

18

Natural follow up to #1318 is to also get `convolve2d` in. This PR is just to get the ball rolling, it's just a super dumb wrapper around `scipy.signal.convolve2d`. Remainings TODO:...

### Description The following raises: ```py import pytensor.tensor as pt import numpy as np x = pt.dmatrix('x') x_val = np.arange(9).reshape((3, 3)) out = pt.subtensor.inc_subtensor(x[:, :2], 10) out.eval({x:x_val}, mode='MLX') ``` Traceback...

Reorganize the `sparse` module

7

## Description Reorganize the existing sparse code to be a bit more logical. In particular, the sparse module now mirrors the tensor module: - `sparse/basic.py` has constructors and utilities -...