flax

updated getting_started in flax docs

#2684

Merged