flax

3f8cec69 - Merge branch 'numpy-convention' of https://github.com/8bitmp3/flax into 8bitmp3-numpy-convention

Merge branch 'numpy-convention' of https://github.com/8bitmp3/flax into 8bitmp3-numpy-convention Manually resolving merge conflicts from 8bitmp3 onp->np PR #798 Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>