.. __init__.py benchmark_cpo_loss.py benchmark_cross_entropy.py benchmark_dpo_loss.py benchmark_embedding.py benchmark_fused_linear_cross_entropy.py benchmark_fused_linear_jsd.py benchmark_geglu.py benchmark_group_norm.py benchmark_jsd.py benchmark_kl_div.py benchmark_layer_norm.py benchmark_orpo_loss.py benchmark_qwen2vl_mrope.py benchmark_rms_norm.py benchmark_rope.py benchmark_simpo_loss.py benchmark_swiglu.py utils.py