GitHub repository: Hemanthhari2000/rax
Description: Rax is a Learning-to-Rank library written in JAX.