Kenzo Clauw

Passing arguments to train multiple models in parallel

4

Hi, I want to perform a gridsearch over different arguments to train multiple models in parallel using optax and flax. My initial idea is to pass an array of learning...