rax (pypi): Learning-to-Rank using JAX.
Keywords: learning-to-rank|jax|ranking