flax

Update getting_started.ipynb and getting_started.md

#2563

Merged