transformers

Fixed a bug calculating cross entropy loss in `JetMoeForCausalLM`

#37830

Merged