https://github.com/amanb2000/Magic_Words/blob/c2a36171be05937a7127a4861030b1ded40f6ea9/magic_words/greedy_forward.py#L274 `Y_to_U[y]['equivalent_u'] = [[u_1], [u_2], ...]` satisfy `y = argmax P(y' | u_i + x_0)` for all `u_i`.