pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Implement Sparse Ops in Numba

Open ricardoV94 opened this issue 2 months ago • 0 comments

We had support for boxing/unboxing Sparse objects in numba, but we couldn't do anything with them.

This PR implements the basic functionality:

  1. CSMProperties (Op that retrieves the attributes from a CSM Matrix)
  2. CSM (Op that rebuilds a Sparse variable from the attributes)
  3. astype()
  4. csr_matrix, csc_matrix sicpy constructor overloads
  5. 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/

ricardoV94 avatar Oct 18 '25 17:10 ricardoV94