torchcell.losses.distributional.dist_param_dim

dist_param_dim(dist: str, num_quantiles: int = 19) → int

Params-per-feature for a config dist value (the head Linear-width multiplier).

energy is 2 because its low-rank factor V is owned by the DistHead (global, [F, k]), not emitted per feature by the structural head.