Preview - rendered architecture diagrams; full validated atlas + dataset coming.
Back to models

U-ViT Diffusion (all-token ViT + U-Net long skip connections for diffusion, d_model=128)

history u vit validation pending

Architecture diagram

Rendered with TorchLens / Graphviz

Open SVG
Unrolled - full
U-ViT Diffusion (all-token ViT + U-Net long skip connections for diffusion, d_model=128) architecture diagram