The routing package hosts affinity metrics and the warm-starting
helper used by TSNAffinityStrategy.
tsn_affinity.routing.metrics — compute_action_affinity,
compute_action_affinity_batch, compute_latent_affinity,
compute_latent_affinity_batch, compute_hybrid_affinity,
compute_diag_stats, extract_obs_latents, normalize_scores,
AffinityMetric.tsn_affinity.routing.warmstarter — MaskWarmstarter.tsn_affinity.routing.metrics::: tsn_affinity.routing.metrics
tsn_affinity.routing.warmstarter::: tsn_affinity.routing.warmstarter