flax

8888c5a8 - Add a comment to MultiOptimizer

Add a comment to MultiOptimizer This is following a question from a user who wasn't sure how to freeze part of a network for the first K steps.