How come I only now stumble on einops? I'm guessing I wasn't paying sufficient attention. https://einops.rocks/