Add simple Flax low-level API Model example to README.md

[sooheon]

No description provided.

[cgarciae]

Thanks @sooheon! Made a small modification, hope you don't mind.

[codecov-io]

Codecov Report

Merging #153 (c03731b) into master (59a32b5) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master     #153   +/-   ##
=======================================
  Coverage   84.90%   84.90%           
=======================================
  Files         131      131           
  Lines        6816     6816           
=======================================
  Hits         5787     5787           
  Misses       1029     1029           

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 59a32b5...c03731b. Read the comment docs.

[sooheon]

Of course not. Thanks for these features! Will be trying out haiku integration at some point as well.