Last released Nov 12, 2024
Parallelization and optimization of Jax models
Last released Sep 10, 2024
Differentiable Multiprocessing for Gradient Descent with JAX
Last released Sep 19, 2023
Galaxy tabulation: HOD counts-in-cells statistics with JAX
Supported by
