PyTorch
API: torch.smm
Strategy: Direct Mapping
JAX (Core)
API: jax.experimental.sparse.spmatmul
Strategy: Direct Mapping
NumPy
API: —
Strategy: Custom / Partial
Keras
API: —
Strategy: Custom / Partial
TensorFlow
API: tf.sparse.sparse_dense_matmul
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Custom / Partial
Flax NNX
API: —
Strategy: Custom / Partial
PaxML / Praxis
API: —
Strategy: Custom / Partial