Update Flax Evolution from Linen to NNX guide

google / flax

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

https://flax.readthedocs.io

Apache License 2.0

6.15k stars 649 forks source link

Closed 8bitmp3 closed 1 month ago

8bitmp3 commented 1 month ago

@IvyZX @cgarciae PTAL