Closed RobertTLange closed 3 years ago
Adresses #24 and adds a subcommand mle init. Furthermore there are a couple of small bug fixes, a lot of refactoring and new helper functions:
mle init
get_jax_os_ready
load_model_ckpt
Adresses #24 and adds a subcommand
mle init
. Furthermore there are a couple of small bug fixes, a lot of refactoring and new helper functions:get_jax_os_ready
helper for setting up JAX environment variables.load_model_ckpt
for smooth reloading of stored checkpoints.