flax

650dcb2b - Merge pull request #2685 from chiamp:flax_docs_extracting_intermediates