flax

updated extracting_intermediates doc

#3063

Merged