Wassim, K. (2025). JAX Distributed Grid Search for Hyperparameter Tuning (Version 0.1.5) [Computer software]