How to resume language modeling training with flax?
I know that Trainer has aneasy wayto resume training. Nonetheless, the mlmscriptfor flax does not contain this functionality. How the training could be resumed with flax in mlm? (restoring weights, optimizer state, etc)