[llama4][auxiliary-loss-free load balancing] update expert_bias without backward hooks#1304

hann-wang wants to merge 2 commits intopytorch:mainpytorch/torchtitan:mainfrom

hann-wang:han/pr_llama4_no_aux_balancinghann-wang/torchtitan:han/pr_llama4_no_aux_balancingCopy head branch name to clipboard