Google выпустила библиотеку для обучения ранжированию TF-Ranking.

В неё включены как самые распространённые алгоритмы и метрики для обучения нейросетей ранжированию, так и собственные разработки, например, метрика LambdaLoss.