Reorganize the sparse module

Author: jessegrabowski

pytensor/link/jax/dispatch/sparse.py

@@ -3,7 +3,7 @@
 
 from pytensor.graph.basic import Constant
 from pytensor.link.jax.dispatch import jax_funcify, jax_typify
-from pytensor.sparse.basic import Dot, StructuredDot
+from pytensor.sparse.math import Dot, StructuredDot
 from pytensor.sparse.type import SparseTensorType
 
 

pytensor/sparse/__init__.py

@@ -1,5 +1,6 @@
 from pytensor.sparse import rewriting, sharedvar
 from pytensor.sparse.basic import *
+from pytensor.sparse.math import *
 from pytensor.sparse.sharedvar import sparse_constructor as shared
 from pytensor.sparse.type import SparseTensorType, _is_sparse