.. experimental model __init__.py auto_model.py cross_entropy.py functional.py fused_linear_cross_entropy.py fused_linear_jsd.py geglu.py group_norm.py jsd.py kl_div.py layer_norm.py monkey_patch.py qwen2vl_mrope.py rms_norm.py rope.py swiglu.py trainer_integration.py