Module weights_geometry 

Replicated checkpoint geometry checked before device allocation.

Functions

expected_replicated_shape

Storage geometry for weights that remain full-width on every TP rank. These are the matrix/vector dimensions consumed by cpu_bind::assemble. Sharded projections are validated by tp::plan_tensor_bytes instead.

validate_replicated_shape