/ src / liger_kernel / chunked_loss / functional.py
functional.py
1  from liger_kernel.chunked_loss.cpo_loss import LigerFusedLinearCPOFunction
2  from liger_kernel.chunked_loss.dpo_loss import LigerFusedLinearDPOFunction
3  from liger_kernel.chunked_loss.orpo_loss import LigerFusedLinearORPOFunction
4  from liger_kernel.chunked_loss.simpo_loss import LigerFusedLinearSimPOFunction
5  
6  liger_fused_linear_orpo = LigerFusedLinearORPOFunction.apply
7  liger_fused_linear_dpo = LigerFusedLinearDPOFunction.apply
8  liger_fused_linear_cpo = LigerFusedLinearCPOFunction.apply
9  liger_fused_linear_simpo = LigerFusedLinearSimPOFunction.apply