import numpy as np


def route(logits, top_k):
    raise NotImplementedError


def load_balancing_loss(logits, top_k):
    raise NotImplementedError