flax
Add some docstrings to the old flax.training.common_utils module.
#2373
Merged
Merged