pytensor
pytensor copied to clipboard
Implement Sparse Ops in Numba
We had support for boxing/unboxing Sparse objects in numba, but we couldn't do anything with them.
This PR implements the basic functionality:
- CSMProperties (Op that retrieves the attributes from a CSM Matrix)
- CSM (Op that rebuilds a Sparse variable from the attributes)
- astype()
- csr_matrix, csc_matrix sicpy constructor overloads
- MULSD (as a POC)
TODO
- [ ] fix conflicts (from #1674)
- [ ] Reintroduce dtype/type on the numba Sparse Matrix
📚 Documentation preview 📚: https://pytensor--1676.org.readthedocs.build/en/1676/