Get the latest tech news

Computing Sharding with Einsum


einsumJanuary 25, 2026 Mental arithmetic in grade school (e.g., memorizing your times tables) is typically justified on the grounds that facility in basic calculations makes it easier to focus on higher-level problems that require being able to do these manipulations. When working on DTensor, I have also found it important to be able to quickly calculate what shardings you get when you do matrix multiplies on sharded tensors.

None

Get the Android app

Or read this on Hacker News

Read more on:

Photo of Einsum

Einsum

Photo of Computing Sharding

Computing Sharding

Related news:

News photo

Einsum in Depth

News photo

Shape Rotation 101: An Intro to Einsum and Jax Transformers

News photo

Einsum for Tensor Manipulation