google / flax

Flax is a neural network library for JAX that is designed for flexibility.

Home Page:https://flax.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Rewrite the Linen Upgrade Guide doc as a HOWTO

marcvanzee opened this issue · comments

#1922 adds a link to the upgrade guide that is currently in a Google Doc, but it is nicer to just put this in our HOWTO format.

We can also add some more lessons learned easily here.