J jax-ml.github.io Sharded matrices and how to multiply them (2025) 1 points lawrenceyan 1 month ago 1 comments
