Module layout 

布局：行宽、badge、分片划分（与真实 checkpoint 结构一致的参数化小实现）。

Structs

Layout

物理布局描述：分片化的定长行存储（如 PLE: 128 shards × [2_500_012, 160] FP8）。

RowExtent

行范围（为批量连续读服务）

Functions

aligned_badge_rows

行宽 -> 尽量使 badge ≥ 4KB 且行数整除性佳的 badge_rows。