v0.2.0

What's Changed

• add colab example to README by @jshin1394 in #113
• add nnx flax update by @jshin1394 in #114
• create metrax/logging package by @jshin1394 in #116
• update pyproject to represent metrax0.2.0 by @jshin1394 in #117

Full Changelog: v0.1.4...v0.2.0