flax

6fb839c6 - Merge pull request #1328 from n2cholas:docstrings