Flax: Trickle down `norm_num_groups`#789

Merged

patrickvonplaten merged 5 commits intohuggingface:mainfrom

akash5474:akash5474-flax-pass-norm-num-groups

Oct 11, 2022

Commits on Oct 10, 2022

using is_flax_available util and created require_flax decorator
akash5474
committed