Similar Items: Cosmologix: Fast, accurate and differentiable distances in the universe with JAX