pytensor

pytensor copied to clipboard

The numpy C-API function has a lot of overhead, as it is made to work with arbitrary types of sequences from Python, plus arguments we don't care about. Also importantly...

ricardoV94

Speed up test times :rocket:

5

If you are motivated to help speed up some tests, we would appreciate it! Here are some of the slowest test times: (9 minutes 17 seconds) ubuntu-latest test py3.13 :...

williambdean

feat: support numba compiled `sort` and `argsort` functions

3

## Description This PR introduces Numba compiled function `sort` and `argsort` for certain kind of sorting algorithms that are supported by Numba (See [this](https://numba.readthedocs.io/en/stable/reference/numpysupported.html)). ## Related Issue - [X] Closes...

victorgarcia98

### Description Reported [here](https://discourse.pymc.io/t/pytensor-errors-from-using-a-sparse-precision-matrix-for-gaussian-data/16773) on the forum. Reproducible example: ```py import pytensor.tensor as pt from pytensor import sparse W_pt = sparse.csc_from_dense(pt.eye(5)) x = pt.dscalar('x') z = (W_pt * x) gz...

jessegrabowski

Split shows up in the graph of Join. It's just a fancy sequence of slice operations. `np.split` is just a thin wrapper that does this: https://github.com/numpy/numpy/blob/e7a123b2d3eca9897843791dd698c1803d9a39c2/numpy/lib/_shape_base_impl.py#L789-L796 If it was not...

ricardoV94

Numpy 2.x C implementation for AdvancedIncSubtensor1

1

### Description In our bump to numpy 2.0 we had to stop supporting the C implementation of this Op https://github.com/pymc-devs/pytensor/pull/1194 This was because we were using this `PyArrayMapIter` functionality that...

ricardoV94

### Description https://github.com/pymc-devs/pytensor/blob/3af923ba1593320cf0a5bfb29a04263406652731/pytensor/compile/function/types.py#L938-L941 I read the discussion, it's not convincing, and even less now that we have other backends besides the C. If someone wants an Op to accept either...

ricardoV94

Make exceptions less verbose by default

11

PyTensor exceptions are very verbose by default. Users often struggle to even find the actual error message. This PR makes exceptions more minimal with a hint to set the relevant...

ricardoV94

Bumps [pypa/cibuildwheel](https://github.com/pypa/cibuildwheel) from 2.23.0 to 2.23.2. Release notes Sourced from pypa/cibuildwheel's releases. v2.23.2 🐛 Workaround an issue with pyodide builds when running cibuildwheel with a Python that was installed via...

dependabot[bot]

Name check on `pt.tensor` is too strict

2

### Description ```py a0 = pt.tensor(name="a0", dtype=floatX, shape=(m,)) ``` Fails with the following error: ```bash a0 = pt.tensor(name="a0", dtype=floatX, shape=(m,)) /home/jesse/mambaforge/envs/pymc-statespace/lib/python3.11/site-packages/pytensor/tensor/type.py:791: in tensor raise ValueError( E ValueError: The first and...

jessegrabowski