flax

c3f73e81 - improve transfer learning guide