flax

5d314528 - Merge pull request #4250 from google:enable-notebook-doctest