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.