flax

5dc65b9d - improve and migrate model_surgery guide