flax

Updated full_eval.ipynb & corrected to constructor

#3059

Merged