ufv_masking#

Mask construction/application for uniform (frame, variations) data.

Masks are static structure and ALWAYS host numpy (np, never xnp) – intentional, required for jit; do not “fix” it (docs/uniform_masks_vs_ranks.md, docs/contributor/uniform_pytree_composition.md). The two ufv_apply_*_masks functions are the frame/variation mask chokepoints and guard the contract with require_concrete_masks, mirroring ut3_apply_masks on the plain layer.

Functions#

ufv_make_frame_masks(up_ranks, down_ranks, left_ranks, ...)

Build the prefix RANK edge masks for a uniform frame. The physical shape is a separate

ufv_apply_frame_masks(data)

Zero the padded ("garbage") regions of the frame supercores via the edge masks. The physical

ufv_apply_variations_masks(data)

Zero the padded ("garbage") regions of the variation supercores via the edge masks. shape_mask