core.nn.metrics# Training-time diagnostics. The reported metrics live in ibl_bwb_eval.metrics. rankme Effective rank of an (entities, dim) embedding: the entropy of its spectrum.