pytensor
pytensor copied to clipboard
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
### Describe the issue: One case where files were missing: #310 The sdist was not finding the correct version number: #293 I think these examples demonstrate the need for some...
### Description This blogpost walks through the logic for 3 different examples: https://www.pymc-labs.com/blog-posts/jax-functions-in-pymc-3-quick-examples/ and shows the logic is always the same: 1. Wrap jitted forward pass in Op 2. Wrap...
### Describe the issue: It seems the JAX linker downcasts index constants to `uint8`? `mode=None` and `mode="NUMBA"` work as expected. Declaring an index variable (`i = pt.lscalar('i'); z = x[i]`)...
### Description This Composite `Op` computes both `Max` and `Argmax` and is returned by default when you call `at.max(...)`. This makes the graphs unnecessarily more complex from the get-go (see...
### Describe the issue: Here is a link to the original discussion at Pymc: https://discourse.pymc.io/t/pytensor-jax-does-not-support-slicing-arrays-with-a-dynamic-slice-length/12163?u=pbaggens The problem is that one cannot use a dynamic slice into a shared variable in...
### Description I just opened #152 using the bug report issue template. The problem is that the "reproducible code example" and "error message" text areas are rendered as python code....
### Describe the issue: These are as expected: ``` from collections,abc import Iterable >>> isinstance(pt.constant([1,2,3]), Iterable) True >>> isinstance(pytensor.shared(np.array([1,2,3])), Iterable) True ``` But a purely symbolic `at.vector()` that doesn't have...
### Description https://github.com/pymc-devs/pytensor/blob/main/pytensor/graph/basic.py#L951 and https://github.com/pymc-devs/pytensor/blob/main/pytensor/graph/basic.py#L1589
### Description Does it make sense to move the converters into pytensor? It will be slightly longer import time, but user experience will be better ```python try: import pandas as...
There are a few links to GitHub Wiki pages which were not migrated by forking. They can be found by search: https://github.com/pymc-devs/pytensor/search?q=wiki ## Tasks + [ ] Check the corresponding...