pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
Hopefully, this will prevent spurious tokenless uploads to codecov
## Description This is very similar in flavor to #742 I'm a bit weary of the fact broadcastability can be redefined by our rewrites, specially after user graph manipulations, because...
## 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) - [...
## Description Implement `pt.pad`, following the `np.pad` API with feature parity. Very preliminary draft, uploading it in this state so I can ask @ricardoV94 to look at the `_linear_ramp_pad` function...
## Description Incorporate ```Rayleigh``` distribution as square root of ```chi2``` distribution with degree of freedom 2. As per the [reference](https://stats.libretexts.org/Bookshelves/Probability_Theory/Probability_Mathematical_Statistics_and_Stochastic_Processes_(Siegrist)/05%3A_Special_Distributions/5.14%3A_The_Rayleigh_Distribution#:~:text=Connections%20to%20the,standard%20Rayleigh%20distribution.). ## Related Issue - [x] Closes #223 - [ ]...
### Description Code: ```python def rayleigh(scale=1.0, *, size=None, **kwargs): return chi2(df=2, size=size, **kwargs) * as_tensor_variable(scale) ``` from: https://github.com/pymc-devs/pytensor/pull/220#issuecomment-1435947464
Add `einsum`
## Description TODO: - [x] Support ellipsis (related to https://github.com/dgasmith/opt_einsum/issues/235) - [x] Exclude broadcastable dims for better perf (JAX does the same) - [x] Handle missing static shape information (default...
### Description [Numpy](https://numpy.org/doc/stable/reference/generated/numpy.pad.html), along with derivative tensor libraries like [jax](https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.pad.html), [torch](https://pytorch.org/docs/2.3/generated/torch.nn.functional.pad.html), etc, all offer padding functions to insert values to the left and right (and top and bottom, etc) to...
### Describe the issue: https://github.com/pymc-devs/pytensor/pull/133 In this PR, the second branch of this conditional: https://github.com/pymc-devs/pytensor/blob/d175203b4e00f48db9c61b68a5f70263a1fbb645/tests/link/jax/test_basic.py#L73-L77 was evidently not updated only because it is never executed in the tests. cc @ricardoV94...
## Description MaxandArgmax Op calculates both maximum and argmax together. With this PR, we aim to have seperate ops for the two operations. ## Related Issue - [x] Closes #334...