flax

a7bb19c9 - Merge pull request #3265 from peterdavidfagan:typo-fix-dropout