flax

1f4b841a - Merge pull request #404 from wrzadkow:fix-optim-readme