liger-kernel-nightly 0.5.2.dev20250109023714__tar.gz → 0.5.2.dev20250115025212__tar.gz

Sign up to get free protection for your applications and to get access to all the features.
Files changed (197) hide show
  1. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/PKG-INFO +6 -6
  2. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/README.md +5 -5
  3. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_cpo_loss.py +8 -15
  4. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_dpo_loss.py +4 -4
  5. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_orpo_loss.py +8 -15
  6. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_simpo_loss.py +8 -15
  7. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/pyproject.toml +1 -1
  8. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/PKG-INFO +6 -6
  9. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/ISSUE_TEMPLATE/bug_report.yaml +0 -0
  10. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/ISSUE_TEMPLATE/feature_request.yaml +0 -0
  11. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/pull_request_template.md +0 -0
  12. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/amd-ci.yml +0 -0
  13. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/nvi-ci.yml +0 -0
  14. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/publish-nightly.yml +0 -0
  15. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/publish-release.yml +0 -0
  16. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.gitignore +0 -0
  17. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/LICENSE +0 -0
  18. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/Makefile +0 -0
  19. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/NOTICE +0 -0
  20. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/__init__.py +0 -0
  21. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/benchmarks_visualizer.py +0 -0
  22. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/data/all_benchmark_data.csv +0 -0
  23. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/__init__.py +0 -0
  24. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_cross_entropy.py +0 -0
  25. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_embedding.py +0 -0
  26. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_fused_linear_cross_entropy.py +0 -0
  27. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_fused_linear_jsd.py +0 -0
  28. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_geglu.py +0 -0
  29. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_group_norm.py +0 -0
  30. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_jsd.py +0 -0
  31. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_kl_div.py +0 -0
  32. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_layer_norm.py +0 -0
  33. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_qwen2vl_mrope.py +0 -0
  34. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_rms_norm.py +0 -0
  35. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_rope.py +0 -0
  36. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_swiglu.py +0 -0
  37. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/utils.py +0 -0
  38. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/fmt-requirements.txt +0 -0
  39. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/modal/tests.py +0 -0
  40. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/modal/tests_bwd.py +0 -0
  41. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/Acknowledgement.md +0 -0
  42. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/CONTRIBUTING.md +0 -0
  43. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/License.md +0 -0
  44. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/banner.GIF +0 -0
  45. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/compose.gif +0 -0
  46. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/e2e-memory.png +0 -0
  47. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/e2e-tps.png +0 -0
  48. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/logo-banner.png +0 -0
  49. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/patch.gif +0 -0
  50. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/post-training.png +0 -0
  51. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/alignment/accelerate_config.yaml +0 -0
  52. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/alignment/run_orpo.py +0 -0
  53. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/README.md +0 -0
  54. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/callback.py +0 -0
  55. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/config/fsdp_config.json +0 -0
  56. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/gemma_7b_mem.png +0 -0
  57. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/gemma_7b_tp.png +0 -0
  58. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/llama_mem_alloc.png +0 -0
  59. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/llama_tps.png +0 -0
  60. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/qwen_mem_alloc.png +0 -0
  61. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/qwen_tps.png +0 -0
  62. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/launch_on_modal.py +0 -0
  63. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/requirements.txt +0 -0
  64. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_benchmarks.sh +0 -0
  65. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_gemma.sh +0 -0
  66. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_llama.sh +0 -0
  67. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_qwen.sh +0 -0
  68. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_qwen2_vl.sh +0 -0
  69. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/training.py +0 -0
  70. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/training_multimodal.py +0 -0
  71. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/README.md +0 -0
  72. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/requirements.txt +0 -0
  73. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/training.py +0 -0
  74. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/README.md +0 -0
  75. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/callback.py +0 -0
  76. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Memory_Stage1_num_head_3.png +0 -0
  77. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Memory_Stage1_num_head_5.png +0 -0
  78. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Memory_Stage2_num_head_3.png +0 -0
  79. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Memory_Stage2_num_head_5.png +0 -0
  80. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Throughput_Stage1_num_head_3.png +0 -0
  81. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Throughput_Stage1_num_head_5.png +0 -0
  82. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Throughput_Stage2_num_head_3.png +0 -0
  83. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/docs/images/Throughput_Stage2_num_head_5.png +0 -0
  84. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/fsdp/acc-fsdp.conf +0 -0
  85. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/medusa_util.py +0 -0
  86. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/requirements.txt +0 -0
  87. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/scripts/llama3_8b_medusa.sh +0 -0
  88. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/train.py +0 -0
  89. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-Apache-2.0 +0 -0
  90. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-AutoAWQ +0 -0
  91. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-Efficient-Cross-Entropy +0 -0
  92. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-llmc +0 -0
  93. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-triton +0 -0
  94. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/setup.cfg +0 -0
  95. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/setup.py +0 -0
  96. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/__init__.py +0 -0
  97. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/README.md +0 -0
  98. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/__init__.py +0 -0
  99. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/cpo_loss.py +0 -0
  100. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/dpo_loss.py +0 -0
  101. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/functional.py +0 -0
  102. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/fused_linear_distillation.py +0 -0
  103. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/fused_linear_preference.py +0 -0
  104. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/orpo_loss.py +0 -0
  105. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/simpo_loss.py +0 -0
  106. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/env_report.py +0 -0
  107. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/__init__.py +0 -0
  108. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/cross_entropy.py +0 -0
  109. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/experimental/embedding.py +0 -0
  110. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/experimental/mm_int8int2.py +0 -0
  111. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/fused_linear_cross_entropy.py +0 -0
  112. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/fused_linear_jsd.py +0 -0
  113. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/geglu.py +0 -0
  114. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/group_norm.py +0 -0
  115. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/jsd.py +0 -0
  116. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/kl_div.py +0 -0
  117. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/layer_norm.py +0 -0
  118. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/qwen2vl_mrope.py +0 -0
  119. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/rms_norm.py +0 -0
  120. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/rope.py +0 -0
  121. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/swiglu.py +0 -0
  122. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/utils.py +0 -0
  123. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/__init__.py +0 -0
  124. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/auto_model.py +0 -0
  125. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/cross_entropy.py +0 -0
  126. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/experimental/embedding.py +0 -0
  127. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/functional.py +0 -0
  128. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/fused_linear_cross_entropy.py +0 -0
  129. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/fused_linear_jsd.py +0 -0
  130. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/geglu.py +0 -0
  131. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/group_norm.py +0 -0
  132. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/jsd.py +0 -0
  133. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/kl_div.py +0 -0
  134. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/layer_norm.py +0 -0
  135. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/__init__.py +0 -0
  136. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/gemma.py +0 -0
  137. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/gemma2.py +0 -0
  138. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/llama.py +0 -0
  139. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mistral.py +0 -0
  140. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mixtral.py +0 -0
  141. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mllama.py +0 -0
  142. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/phi3.py +0 -0
  143. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/qwen2.py +0 -0
  144. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/qwen2_vl.py +0 -0
  145. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/monkey_patch.py +0 -0
  146. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/qwen2vl_mrope.py +0 -0
  147. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/rms_norm.py +0 -0
  148. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/rope.py +0 -0
  149. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/swiglu.py +0 -0
  150. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer/__init__.py +0 -0
  151. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer/orpo_trainer.py +0 -0
  152. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer_integration.py +0 -0
  153. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/triton/__init__.py +0 -0
  154. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/triton/monkey_patch.py +0 -0
  155. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/utils.py +0 -0
  156. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/SOURCES.txt +0 -0
  157. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/dependency_links.txt +0 -0
  158. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/requires.txt +0 -0
  159. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/top_level.txt +0 -0
  160. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/__init__.py +0 -0
  161. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/__init__.py +0 -0
  162. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_cpo_loss.py +0 -0
  163. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_dpo_loss.py +0 -0
  164. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_orpo_loss.py +0 -0
  165. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_simpo_loss.py +0 -0
  166. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/conftest.py +0 -0
  167. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/__init__.py +0 -0
  168. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models.py +0 -0
  169. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models_multimodal.py +0 -0
  170. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models_with_logits.py +0 -0
  171. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/fake_configs/Qwen/Qwen2-VL-7B-Instruct/tokenizer_config.json +0 -0
  172. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/fake_configs/meta-llama/Llama-3.2-11B-Vision-Instruct/tokenizer_config.json +0 -0
  173. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/scripts/generate_tokenized_dataset.py +0 -0
  174. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare.txt +0 -0
  175. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare_tokenized/data-00000-of-00001.arrow +0 -0
  176. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare_tokenized/dataset_info.json +0 -0
  177. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare_tokenized/state.json +0 -0
  178. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_auto_model.py +0 -0
  179. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_cross_entropy.py +0 -0
  180. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_embedding.py +0 -0
  181. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_fused_linear_cross_entropy.py +0 -0
  182. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_fused_linear_jsd.py +0 -0
  183. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_geglu.py +0 -0
  184. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_group_norm.py +0 -0
  185. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_jsd.py +0 -0
  186. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_kl_div.py +0 -0
  187. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_layer_norm.py +0 -0
  188. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_mm_int8int2.py +0 -0
  189. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_monkey_patch.py +0 -0
  190. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_qwen2vl_mrope.py +0 -0
  191. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_rms_norm.py +0 -0
  192. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_rope.py +0 -0
  193. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_swiglu.py +0 -0
  194. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_trainer_integration.py +0 -0
  195. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_transformers.py +0 -0
  196. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/triton/test_triton_monkey_patch.py +0 -0
  197. {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: liger_kernel_nightly
3
- Version: 0.5.2.dev20250109023714
3
+ Version: 0.5.2.dev20250115025212
4
4
  Summary: Efficient Triton kernels for LLM Training
5
5
  License: BSD 2-CLAUSE LICENSE
6
6
  Copyright 2024 LinkedIn Corporation
@@ -105,8 +105,8 @@ Requires-Dist: seaborn; extra == "dev"
105
105
 
106
106
  <details>
107
107
  <summary>Latest News 🔥</summary>
108
-
109
- - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
108
+
109
+ - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
110
110
  - [2024/12/5] We release LinkedIn Engineering Blog - [Liger-Kernel: Empowering an open source ecosystem of Triton Kernels for Efficient LLM Training](https://www.linkedin.com/blog/engineering/open-source/liger-kernel-open-source-ecosystem-for-efficient-llm-training)
111
111
  - [2024/11/6] We release [v0.4.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.4.0): Full AMD support, Tech Report, Modal CI, Llama-3.2-Vision!
112
112
  - [2024/10/21] We have released the tech report of Liger Kernel on Arxiv: https://arxiv.org/pdf/2410.10989
@@ -299,7 +299,7 @@ loss.backward()
299
299
  | Gemma1 | `liger_kernel.transformers.apply_liger_kernel_to_gemma` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
300
300
  | Gemma2 | `liger_kernel.transformers.apply_liger_kernel_to_gemma2` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
301
301
  | Qwen2, Qwen2.5, & QwQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
302
- | Qwen2-VL | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
302
+ | Qwen2-VL, & QVQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
303
303
  | Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
304
304
 
305
305
 
@@ -353,11 +353,11 @@ loss.backward()
353
353
  - [License Information](https://github.com/linkedin/Liger-Kernel/blob/main/docs/License.md)
354
354
 
355
355
  ## Sponsorship and Collaboration
356
-
356
+
357
357
  - [AMD](https://www.amd.com/en.html): Providing AMD GPUs for our AMD CI.
358
358
  - [Intel](https://www.intel.com/): Providing Intel GPUs for our Intel CI.
359
359
  - [Modal](https://modal.com/): Free 3000 credits from GPU MODE IRL for our NVIDIA CI.
360
- - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
360
+ - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
361
361
  - [HuggingFace](https://huggingface.co/): Integrating Liger Kernel into Hugging Face Transformers and TRL.
362
362
  - [Lightning AI](https://lightning.ai/): Integrating Liger Kernel into Lightning Thunder.
363
363
  - [Axolotl](https://axolotl.ai/): Integrating Liger Kernel into Axolotl.
@@ -59,8 +59,8 @@
59
59
 
60
60
  <details>
61
61
  <summary>Latest News 🔥</summary>
62
-
63
- - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
62
+
63
+ - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
64
64
  - [2024/12/5] We release LinkedIn Engineering Blog - [Liger-Kernel: Empowering an open source ecosystem of Triton Kernels for Efficient LLM Training](https://www.linkedin.com/blog/engineering/open-source/liger-kernel-open-source-ecosystem-for-efficient-llm-training)
65
65
  - [2024/11/6] We release [v0.4.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.4.0): Full AMD support, Tech Report, Modal CI, Llama-3.2-Vision!
66
66
  - [2024/10/21] We have released the tech report of Liger Kernel on Arxiv: https://arxiv.org/pdf/2410.10989
@@ -253,7 +253,7 @@ loss.backward()
253
253
  | Gemma1 | `liger_kernel.transformers.apply_liger_kernel_to_gemma` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
254
254
  | Gemma2 | `liger_kernel.transformers.apply_liger_kernel_to_gemma2` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
255
255
  | Qwen2, Qwen2.5, & QwQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
256
- | Qwen2-VL | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
256
+ | Qwen2-VL, & QVQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
257
257
  | Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
258
258
 
259
259
 
@@ -307,11 +307,11 @@ loss.backward()
307
307
  - [License Information](https://github.com/linkedin/Liger-Kernel/blob/main/docs/License.md)
308
308
 
309
309
  ## Sponsorship and Collaboration
310
-
310
+
311
311
  - [AMD](https://www.amd.com/en.html): Providing AMD GPUs for our AMD CI.
312
312
  - [Intel](https://www.intel.com/): Providing Intel GPUs for our Intel CI.
313
313
  - [Modal](https://modal.com/): Free 3000 credits from GPU MODE IRL for our NVIDIA CI.
314
- - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
314
+ - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
315
315
  - [HuggingFace](https://huggingface.co/): Integrating Liger Kernel into Hugging Face Transformers and TRL.
316
316
  - [Lightning AI](https://lightning.ai/): Integrating Liger Kernel into Lightning Thunder.
317
317
  - [Axolotl](https://axolotl.ai/): Integrating Liger Kernel into Axolotl.
@@ -11,7 +11,6 @@ from utils import _test_memory
11
11
  from utils import parse_benchmark_script_args
12
12
  from utils import run_benchmarks
13
13
 
14
- from liger_kernel.chunked_loss.cpo_loss import LigerFusedLinearCPOFunction
15
14
  from liger_kernel.utils import infer_device
16
15
 
17
16
  device = infer_device()
@@ -27,7 +26,8 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../.
27
26
  def bench_memory_fused_linear_cpo_loss(
28
27
  input: SingleBenchmarkRunInput,
29
28
  ) -> SingleBenchmarkRunOutput:
30
- from test.chunked_loss.test_cpo_loss import LigerLMHeadCPO, TorchLMHeadCPO
29
+ from test.chunked_loss.test_cpo_loss import LigerLMHeadCPO
30
+ from test.chunked_loss.test_cpo_loss import TorchLMHeadCPO
31
31
 
32
32
  B = input.x
33
33
  T = input.extra_benchmark_config["T"]
@@ -36,12 +36,8 @@ def bench_memory_fused_linear_cpo_loss(
36
36
  dtype = input.extra_benchmark_config["dtype"]
37
37
  provider = input.kernel_provider
38
38
 
39
- torch_lm_head_cpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(
40
- device
41
- )(x, target)[0]
42
- liger_lm_head_cpo = lambda x, target: LigerLMHeadCPO(H=H, V=V, dtype=dtype).to(
43
- device
44
- )(x, target)[0]
39
+ torch_lm_head_cpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
40
+ liger_lm_head_cpo = lambda x, target: LigerLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
45
41
 
46
42
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
47
43
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -72,7 +68,8 @@ def bench_memory_fused_linear_cpo_loss(
72
68
  def bench_speed_fused_linear_cpo_loss(
73
69
  input: SingleBenchmarkRunInput,
74
70
  ) -> SingleBenchmarkRunOutput:
75
- from test.chunked_loss.test_cpo_loss import LigerLMHeadCPO, TorchLMHeadCPO
71
+ from test.chunked_loss.test_cpo_loss import LigerLMHeadCPO
72
+ from test.chunked_loss.test_cpo_loss import TorchLMHeadCPO
76
73
 
77
74
  B = input.x
78
75
  T = input.extra_benchmark_config["T"]
@@ -82,12 +79,8 @@ def bench_speed_fused_linear_cpo_loss(
82
79
  provider = input.kernel_provider
83
80
  mode = input.kernel_operation_mode
84
81
 
85
- torch_lm_head_cpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(
86
- device
87
- )(x, target)[0]
88
- liger_lm_head_cpo = lambda x, target: LigerLMHeadCPO(H=H, V=V, dtype=dtype).to(
89
- device
90
- )(x, target)[0]
82
+ torch_lm_head_cpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
83
+ liger_lm_head_cpo = lambda x, target: LigerLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
91
84
 
92
85
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
93
86
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -4,7 +4,6 @@ import sys
4
4
  import torch
5
5
  import triton
6
6
 
7
- from test.chunked_loss.test_dpo_loss import HF_DPO_Loss
8
7
  from utils import QUANTILES
9
8
  from utils import SingleBenchmarkRunInput
10
9
  from utils import SingleBenchmarkRunOutput
@@ -12,7 +11,6 @@ from utils import _test_memory
12
11
  from utils import parse_benchmark_script_args
13
12
  from utils import run_benchmarks
14
13
 
15
- from liger_kernel.chunked_loss.dpo_loss import LigerFusedLinearDPOFunction
16
14
  from liger_kernel.utils import infer_device
17
15
 
18
16
  device = infer_device()
@@ -21,7 +19,8 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../.
21
19
 
22
20
 
23
21
  def bench_memory_dpo_loss(input: SingleBenchmarkRunInput) -> SingleBenchmarkRunOutput:
24
- from test.chunked_loss.test_dpo_loss import LigerLMHeadDPO, TorchLMHeadDPO
22
+ from test.chunked_loss.test_dpo_loss import LigerLMHeadDPO
23
+ from test.chunked_loss.test_dpo_loss import TorchLMHeadDPO
25
24
 
26
25
  B = input.x
27
26
  T = input.extra_benchmark_config["T"]
@@ -70,7 +69,8 @@ def bench_memory_dpo_loss(input: SingleBenchmarkRunInput) -> SingleBenchmarkRunO
70
69
 
71
70
 
72
71
  def bench_speed_dpo_loss(input: SingleBenchmarkRunInput) -> SingleBenchmarkRunOutput:
73
- from test.chunked_loss.test_dpo_loss import LigerLMHeadDPO, TorchLMHeadDPO
72
+ from test.chunked_loss.test_dpo_loss import LigerLMHeadDPO
73
+ from test.chunked_loss.test_dpo_loss import TorchLMHeadDPO
74
74
 
75
75
  B = input.x
76
76
  T = input.extra_benchmark_config["T"]
@@ -11,7 +11,6 @@ from utils import _test_memory
11
11
  from utils import parse_benchmark_script_args
12
12
  from utils import run_benchmarks
13
13
 
14
- from liger_kernel.chunked_loss.orpo_loss import LigerFusedLinearORPOFunction
15
14
  from liger_kernel.utils import infer_device
16
15
 
17
16
  device = infer_device()
@@ -27,7 +26,8 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../.
27
26
  def bench_memory_fused_linear_orpo_loss(
28
27
  input: SingleBenchmarkRunInput,
29
28
  ) -> SingleBenchmarkRunOutput:
30
- from test.chunked_loss.test_orpo_loss import LigerLMHeadORPO, TorchLMHeadORPO
29
+ from test.chunked_loss.test_orpo_loss import LigerLMHeadORPO
30
+ from test.chunked_loss.test_orpo_loss import TorchLMHeadORPO
31
31
 
32
32
  B = input.x
33
33
  T = input.extra_benchmark_config["T"]
@@ -36,12 +36,8 @@ def bench_memory_fused_linear_orpo_loss(
36
36
  dtype = input.extra_benchmark_config["dtype"]
37
37
  provider = input.kernel_provider
38
38
 
39
- torch_lm_head_orpo = lambda x, target: TorchLMHeadORPO(H=H, V=V, dtype=dtype).to(
40
- device
41
- )(x, target)[0]
42
- liger_lm_head_orpo = lambda x, target: LigerLMHeadORPO(H=H, V=V, dtype=dtype).to(
43
- device
44
- )(x, target)[0]
39
+ torch_lm_head_orpo = lambda x, target: TorchLMHeadORPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
40
+ liger_lm_head_orpo = lambda x, target: LigerLMHeadORPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
45
41
 
46
42
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
47
43
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -73,7 +69,8 @@ def bench_memory_fused_linear_orpo_loss(
73
69
  def bench_speed_fused_linear_orpo_loss(
74
70
  input: SingleBenchmarkRunInput,
75
71
  ) -> SingleBenchmarkRunOutput:
76
- from test.chunked_loss.test_orpo_loss import LigerLMHeadORPO, TorchLMHeadORPO
72
+ from test.chunked_loss.test_orpo_loss import LigerLMHeadORPO
73
+ from test.chunked_loss.test_orpo_loss import TorchLMHeadORPO
77
74
 
78
75
  B = input.x
79
76
  T = input.extra_benchmark_config["T"]
@@ -83,12 +80,8 @@ def bench_speed_fused_linear_orpo_loss(
83
80
  provider = input.kernel_provider
84
81
  mode = input.kernel_operation_mode
85
82
 
86
- torch_lm_head_orpo = lambda x, target: TorchLMHeadORPO(H=H, V=V, dtype=dtype).to(
87
- device
88
- )(x, target)[0]
89
- liger_lm_head_orpo = lambda x, target: LigerLMHeadORPO(H=H, V=V, dtype=dtype).to(
90
- device
91
- )(x, target)[0]
83
+ torch_lm_head_orpo = lambda x, target: TorchLMHeadORPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
84
+ liger_lm_head_orpo = lambda x, target: LigerLMHeadORPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
92
85
 
93
86
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
94
87
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -11,7 +11,6 @@ from utils import _test_memory
11
11
  from utils import parse_benchmark_script_args
12
12
  from utils import run_benchmarks
13
13
 
14
- from liger_kernel.chunked_loss.simpo_loss import LigerFusedLinearSimPOFunction
15
14
  from liger_kernel.utils import infer_device
16
15
 
17
16
  device = infer_device()
@@ -27,7 +26,8 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../.
27
26
  def bench_memory_fused_linear_simpo_loss(
28
27
  input: SingleBenchmarkRunInput,
29
28
  ) -> SingleBenchmarkRunOutput:
30
- from test.chunked_loss.test_simpo_loss import LigerLMHeadSimPO, TorchLMHeadCPO
29
+ from test.chunked_loss.test_simpo_loss import LigerLMHeadSimPO
30
+ from test.chunked_loss.test_simpo_loss import TorchLMHeadCPO
31
31
 
32
32
  B = input.x
33
33
  T = input.extra_benchmark_config["T"]
@@ -36,12 +36,8 @@ def bench_memory_fused_linear_simpo_loss(
36
36
  dtype = input.extra_benchmark_config["dtype"]
37
37
  provider = input.kernel_provider
38
38
 
39
- torch_lm_head_simpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(
40
- device
41
- )(x, target)[0]
42
- liger_lm_head_simpo = lambda x, target: LigerLMHeadSimPO(H=H, V=V, dtype=dtype).to(
43
- device
44
- )(x, target)[0]
39
+ torch_lm_head_simpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
40
+ liger_lm_head_simpo = lambda x, target: LigerLMHeadSimPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
45
41
 
46
42
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
47
43
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -72,7 +68,8 @@ def bench_memory_fused_linear_simpo_loss(
72
68
  def bench_speed_fused_linear_simpo_loss(
73
69
  input: SingleBenchmarkRunInput,
74
70
  ) -> SingleBenchmarkRunOutput:
75
- from test.chunked_loss.test_simpo_loss import LigerLMHeadSimPO, TorchLMHeadCPO
71
+ from test.chunked_loss.test_simpo_loss import LigerLMHeadSimPO
72
+ from test.chunked_loss.test_simpo_loss import TorchLMHeadCPO
76
73
 
77
74
  B = input.x
78
75
  T = input.extra_benchmark_config["T"]
@@ -82,12 +79,8 @@ def bench_speed_fused_linear_simpo_loss(
82
79
  provider = input.kernel_provider
83
80
  mode = input.kernel_operation_mode
84
81
 
85
- torch_lm_head_simpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(
86
- device
87
- )(x, target)[0]
88
- liger_lm_head_simpo = lambda x, target: LigerLMHeadSimPO(H=H, V=V, dtype=dtype).to(
89
- device
90
- )(x, target)[0]
82
+ torch_lm_head_simpo = lambda x, target: TorchLMHeadCPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
83
+ liger_lm_head_simpo = lambda x, target: LigerLMHeadSimPO(H=H, V=V, dtype=dtype).to(device)(x, target)[0]
91
84
 
92
85
  _input = torch.randn(B, T, H, requires_grad=True, dtype=dtype, device=device)
93
86
  target = torch.randint(V, (B, T), dtype=torch.long, device=device)
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "liger_kernel_nightly"
7
- version = "0.5.2.dev20250109023714"
7
+ version = "0.5.2.dev20250115025212"
8
8
  description = "Efficient Triton kernels for LLM Training"
9
9
  urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
10
10
  readme = { file = "README.md", content-type = "text/markdown" }
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: liger_kernel_nightly
3
- Version: 0.5.2.dev20250109023714
3
+ Version: 0.5.2.dev20250115025212
4
4
  Summary: Efficient Triton kernels for LLM Training
5
5
  License: BSD 2-CLAUSE LICENSE
6
6
  Copyright 2024 LinkedIn Corporation
@@ -105,8 +105,8 @@ Requires-Dist: seaborn; extra == "dev"
105
105
 
106
106
  <details>
107
107
  <summary>Latest News 🔥</summary>
108
-
109
- - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
108
+
109
+ - [2024/12/11] We release [v0.5.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.5.0): 80% more memory efficient post training losses (DPO, ORPO, CPO, etc)!
110
110
  - [2024/12/5] We release LinkedIn Engineering Blog - [Liger-Kernel: Empowering an open source ecosystem of Triton Kernels for Efficient LLM Training](https://www.linkedin.com/blog/engineering/open-source/liger-kernel-open-source-ecosystem-for-efficient-llm-training)
111
111
  - [2024/11/6] We release [v0.4.0](https://github.com/linkedin/Liger-Kernel/releases/tag/v0.4.0): Full AMD support, Tech Report, Modal CI, Llama-3.2-Vision!
112
112
  - [2024/10/21] We have released the tech report of Liger Kernel on Arxiv: https://arxiv.org/pdf/2410.10989
@@ -299,7 +299,7 @@ loss.backward()
299
299
  | Gemma1 | `liger_kernel.transformers.apply_liger_kernel_to_gemma` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
300
300
  | Gemma2 | `liger_kernel.transformers.apply_liger_kernel_to_gemma2` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
301
301
  | Qwen2, Qwen2.5, & QwQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
302
- | Qwen2-VL | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
302
+ | Qwen2-VL, & QVQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
303
303
  | Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
304
304
 
305
305
 
@@ -353,11 +353,11 @@ loss.backward()
353
353
  - [License Information](https://github.com/linkedin/Liger-Kernel/blob/main/docs/License.md)
354
354
 
355
355
  ## Sponsorship and Collaboration
356
-
356
+
357
357
  - [AMD](https://www.amd.com/en.html): Providing AMD GPUs for our AMD CI.
358
358
  - [Intel](https://www.intel.com/): Providing Intel GPUs for our Intel CI.
359
359
  - [Modal](https://modal.com/): Free 3000 credits from GPU MODE IRL for our NVIDIA CI.
360
- - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
360
+ - [EmbeddedLLM](https://embeddedllm.com/): Making Liger Kernel run fast and stable on AMD.
361
361
  - [HuggingFace](https://huggingface.co/): Integrating Liger Kernel into Hugging Face Transformers and TRL.
362
362
  - [Lightning AI](https://lightning.ai/): Integrating Liger Kernel into Lightning Thunder.
363
363
  - [Axolotl](https://axolotl.ai/): Integrating Liger Kernel into Axolotl.