pytensor
pytensor copied to clipboard
pymc-devs
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
### Description AdvancedSubtensor1 and AdvancedIncSubtensor1 are specializations of AdvancedSubtensor and AdvancedIncSubtensor that only allow one vector of integer indexing along the first dimension. They exist to provide a C-implementation of...
`dvector` hardcodes `shape=(None,)`. Would be nicer to make it a partial of `vector` with the fixed dtype to still allow providing a specific static shape if the user wants to...
https://github.com/pymc-devs/pytensor/blob/5d4b0c4b9a1e478dda48e912ee708a9e557e9343/pytensor/tensor/random/basic.py#L897-L903 It is not allowed to broadcast with mu so it should be aware of its shape. And should probably be eye(mu.shape[-1]), not ones. Or we can just remove it,...
## Description Added more robust docstrings and removed the Elemwise docstring that was appended to each function. ## Related Issue - [x] Closes #292 - [ ] Related to #...
### Description NumPy 2.0 introduced some breaking changes in APIs, which break code and make tests unable to run. One example is np.obj2sctype(dtype) used in pytensor/tensor/type.py:105, which was removed in...
### Description The easiest way to give PyTensor users global (not fine-grained) control over CPU/GPU for the PyTorch backend would be the `set_default_device`/`with_device`. However, this may bee too slow, according...
### Description https://github.com/pymc-devs/pytensor/blob/0e29d76b3122ba4fdedf0531005a4c0ba178d270/pytensor/link/jax/dispatch/basic.py#L22-L28 (And now from copying the approach also the PyTorch Linker)
## Description ## What is broken / not supported by Numba: - float16: https://github.com/numba/numba/issues/4402 - Deepcopy / copying protected outputs: https://github.com/pymc-devs/pytensor/issues/50 - Sparse operations ## Related Issue - [ ]...
### Please describe the purpose of filing this issue Equivalent symbolic methods to those are missing. The Numpy `quantile` and `percentile` methods have too many options for the `interpolation` argument,...
## Description So here, I aim to replace the output of a graph of ```x.[1]``` or ```x@[1]``` with x. I am not quite sure if this is the correct way...
