Cosmologix: Fast, accurate and differentiable distances in the universe with JAX (Journal of Open Source Software, 2026)