view article Article Shape Rotation 101: An Intro to Einsum and Jax Transformers By dejavucoder • Jun 22 • 2