Data & analyticsDiRe - JAX: A JAX based Dimensionality Reduction Algorithm for Large-scale DataResearch paper Jun 2025