* generalize decay_mask_fn to find all layernorm params * fixup * generalising decay_mask_fn
Attach a file by drag & drop or click to upload
