flax

c33df578 - Renames get_param to get_initial_params in seq2seq and mnist Linen examples