flax

2f76a92d - Merge pull request #3765 from chiamp:docstring_init_apply