Dsmm

Performs a matrix multiplication of the sparse matrix input and the dense matrix mat2.

Abstract Signature:

Dsmm(input: Tensor, mat2: Tensor)

PyTorch

API: torch.dsmm
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.experimental.sparse.bcoo_spdot_general
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.sparse.sparse_dense_matmul
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.experimental.sparse.bcoo_spdot_general
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.experimental.sparse.bcoo_spdot_general
Strategy: Direct Mapping
Official Docs ↗