Tensor Formula Viz

Turn a formula or tensor-code path into one dense, slide-ready figure with three zones:

1. Top — formula: clean mathematical definition, without shape underbraces.
2. Middle — computation: colored blocks, stripes, slices, operators, and communication.
3. Bottom — meaning: compact explanation of every shape dimension and non-obvious operation.

Workflow

1. Reduce the input to one primary computation path. Omit equivalent objectives, diagnostics, and secondary metrics unless requested.
2. Build a shape-and-semantics ledger: symbol/code name, semantic kind, dtype/domain, global and local shape, axis meanings, range when meaningful, shape-equivalence class, producer/consumer, and contracted/broadcast/reduced axes.
3. Choose the smallest visual grammar that exposes the mechanism.
4. Reserve non-overlapping stage lanes and object bounding boxes, then draw the three-zone layout.
5. Render, inspect, and correct both mathematical and visual alignment.

For code input, trace concrete matmul, einsum, reshape/view, transpose/permute, concat, broadcast, and collective operations. Preserve code variable names when useful; state any inferred shapes or conventions.

Choose the visual grammar