flax

bbcf96c9 - Merge pull request #2563 from chiamp:getting_started_notebook