Last released Dec 11, 2025
Parallelization and optimization of Jax models
Last released Jan 7, 2025
Galaxy tabulation: HOD counts-in-cells statistics with JAX
Last released Sep 10, 2024
Differentiable Multiprocessing for Gradient Descent with JAX
Supported by
