pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
Implement indexing ops in pytorch - [x] Subtensor - [ ] IncSubtensor - [x] AdvancedSubtensor[1] - [ ] AdvancedIncSubtensor[1] ## Description ## Related Issue - [x] Related to #821 ##...
## Description Simplify the function class in pytensor. - [x] Remove the `output_subset` kwargs at call time of class `Function` as no one is using it. - [ ] Check...
### Describe the issue: ``` ERROR (pytensor.graph.rewriting.basic): Rewrite failure due to: constant_folding ERROR (pytensor.graph.rewriting.basic): node: Cast{float64}(0) ERROR (pytensor.graph.rewriting.basic): TRACEBACK: ERROR (pytensor.graph.rewriting.basic): Traceback (most recent call last): File "/Users/muntazirabidi/miniforge3/envs/pymc/lib/python3.12/site-packages/pytensor/graph/rewriting/basic.py", line 1919,...
## Description * First commit: adding a `strict=True` argument to all zips when it doesn't produce mistakes in the test suite (464 of them), and `strict=False` to the others (28...
```python import numpy as np from pytensor import function from pytensor.xtensor.basic import add, exp from pytensor.xtensor.type import xtensor x = xtensor("x", dims=("city",), shape=(None,)) y = xtensor("y", dims=("country",), shape=(4,)) z =...
## Description Herein I aim to convert all: - X[0:-1:1] -> X[:-1] - X[0:-1] -> X[:-1] - X[:-1] -> X[:-1] ## Related Issue - [x] Closes #58 - [ ]...
### Before ```python import pytensor.tensor as pt # Need to a = pt.vector("a", shape=(2, )) b = pt.vector("b", shape=(3, )) # a + b fails due to broadcasting # Transpose...
## Description Added ```FutureWarnings``` whenever ```test_value``` is accessed and also added ```pytest.warnings``` when tests use it. ## Related Issue - [ ] Closes # - [x] Related to #447 ##...
### Please describe the purpose of filing this issue Not a drastic improvement by any means, but something we can keep in mind: ``` reduce(at.concatenate(*tensors)) -> reduce(reduce(tensor) for tensor in...
### Description IfElse is only lazy in the default backend because the function virtual machine handles it (via the "lazy" attribute"). In Numba/JAX it currently does nothing, because it receives...