search

google / flax

Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.78k stars 610 forks source link

Add NNX surgery guide #4005

Closed IvyZX closed 6 days ago

IvyZX commented 2 weeks ago

Add a guide for common model surgery scenarios and how to use NNX state and modules to fluently surgery around.

review-notebook-app[bot] commented 2 weeks ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

codecov-commenter commented 1 week ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 0.00%. Comparing base (31adb00) to head (7eeaa7b). Report is 26 commits behind head on main.

Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #4005 +/- ## ====================================== Coverage 0.00% 0.00% ====================================== Files 106 107 +1 Lines 13582 13689 +107 ====================================== - Misses 13582 13689 +107 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.