Add `linalg.block_diag` and sparse equivalent #576

Merged

jessegrabowski merged 16 commits intopymc-devs:mainfrom

jessegrabowski:block-diag

Jan 7, 2024

Commits on Jan 6, 2024

Copy `block_diag` and support functions from `pymc.math`
jessegrabowski
committed
Evaluate output in sphinx code example

jessegrabowski
and
ricardoV94
authored
Test type equivalence with `isinstance` instead of `==`

jessegrabowski
and
ricardoV94
authored
Typo in test function
jessegrabowski
committed
Split `block_diag` into sparse and dense version
jessegrabowski
committed
Use `as_sparse_or_tensor_variable` in `SparseBlockDiagonalMatrix` to allow sparse matrix inputs to `pytensor.sparse.block_diag`
jessegrabowski
committed
Test sparse and dense inputs to `pytensor.sparse.block_diag`
jessegrabowski
committed
Add numba overload for `pytensor.tensor.slinalg.block_diag`
jessegrabowski
committed
add jax overload for `pytensor.tensor.slinalg.block_diag`
jessegrabowski
committed
Move stand-alone `block_diag_grad` function into `grad` method
jessegrabowski
committed
Add `format` prop to `SparseBlockDiagonalMatrix`
jessegrabowski
committed
Use `compare_numba_and_py` in `numba\test_slinalg.py::test_block_diag`
jessegrabowski
committed
Add support for Blockwise to `slinalg.block_diag`
jessegrabowski
committed

Commits on Jan 7, 2024