einsum visualization
llm-vis