Sharded matrices and how to multiply them (2025)(jax-ml.github.io)1 points by lawrenceyan 12 days ago | 1 comment