Add balance loss in atorch moe example

[skydoorkai]

Add balance loss in moe example.

We add use aux loss to router in moe example, similar to aux loss implementation in megatron moe.