https://www.graphcore.ai/posts/accelerating-block-sparse-matrix-multiplication-on-ipus-with-popsparse
Accelerating Block Sparse Matrix Multiplication on IPUs with PopSparse
Introducing PopSparse, a library that enables fast sparse operations on the Graphcore IPU.
block sparseacceleratingipus
https://docs.jax.dev/en/latest/pallas/tpu/sparse.html
Scalar Prefetch and Block-Sparse Computation — JAX documentation
block sparsejax documentation