liger-kernel-nightly 0.6.2.dev20251005233549__tar.gz → 0.6.4.dev20260121000828__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of liger-kernel-nightly might be problematic. Click here for more details.
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/amd-ci.yml +4 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/benchmark.yml +14 -10
- liger_kernel_nightly-0.6.4.dev20260121000828/.github/workflows/docs.yml +64 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/intel-ci.yml +3 -3
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/nvi-ci.yml +6 -6
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/publish-nightly.yml +2 -2
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/workflows/publish-release.yml +2 -2
- liger_kernel_nightly-0.6.4.dev20260121000828/.pre-commit-config.yaml +10 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/PKG-INFO +13 -3
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/README.md +11 -2
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/data/all_benchmark_data.csv +240 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_cross_entropy.py +4 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_fused_linear_cross_entropy.py +25 -19
- liger_kernel_nightly-0.6.4.dev20260121000828/benchmark/scripts/benchmark_poly_norm.py +197 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/benchmark/scripts/benchmark_tiled_mlp.py +397 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_tvd.py +7 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/dev/modal/tests.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/dev/modal/tests_bwd.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/acknowledgement.md +0 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/contributing.md +45 -48
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/pyproject.toml +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/setup.py +21 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/cosine_similarity_loss.py +20 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/fused_linear_distillation.py +23 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/fused_linear_ppo.py +21 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/grpo_loss.py +8 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/jsd_loss.py +39 -11
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/__init__.py +141 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/README.md +151 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/__init__.py +13 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/__init__.py +5 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ascend-ub-manager-design.md +485 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/__init__.py +49 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/geglu.py +266 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/qwen2vl_mrope.py +285 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/rope.py +290 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/swiglu.py +142 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ops/tvd.py +221 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/_ascend/ub_manager.py +349 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/backends/registry.py +61 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/cross_entropy.py +130 -64
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/dyt.py +5 -2
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/fused_add_rms_norm.py +21 -23
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/fused_linear_cross_entropy.py +45 -13
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/geglu.py +5 -3
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/group_norm.py +12 -8
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/grpo_loss.py +3 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/kl_div.py +8 -11
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/layer_norm.py +89 -69
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/poly_norm.py +384 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/rms_norm.py +149 -71
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/ops/tiled_mlp.py +136 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/utils.py +25 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/__init__.py +36 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/auto_model.py +21 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/cross_entropy.py +9 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/dyt.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/experimental/embedding.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/functional.py +48 -25
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/fused_add_rms_norm.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/fused_linear_cross_entropy.py +9 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/fused_linear_jsd.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/fused_neighborhood_attention.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/geglu.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/group_norm.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/grpo_loss.py +57 -2
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/jsd.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/kl_div.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/layer_norm.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/llama4_rope.py +1 -1
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/exaone4.py +136 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/falcon_h1.py +122 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/gemma.py +19 -7
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/gemma2.py +25 -10
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/gemma3.py +62 -18
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/glm4.py +18 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/glm4v.py +18 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/glm4v_moe.py +25 -5
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/gpt_oss.py +211 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/hunyuan_v1.py +134 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/internvl.py +12 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/llama.py +16 -6
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/llama4.py +18 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/llava.py +18 -6
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/loss_utils.py +37 -3
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/mistral.py +17 -7
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/mixtral.py +24 -9
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/mllama.py +14 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/olmo2.py +18 -5
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/olmo3.py +142 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/output_classes.py +147 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/paligemma.py +42 -5
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/phi3.py +16 -8
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/qwen2.py +18 -4
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/qwen2_5_vl.py +163 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/qwen2_vl.py +24 -7
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/qwen3.py +22 -6
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/qwen3_moe.py +27 -7
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/qwen3_next.py +146 -0
- liger_kernel_nightly-0.6.2.dev20251005233549/src/liger_kernel/transformers/model/qwen2_5_vl.py → liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/qwen3_vl.py +17 -17
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/qwen3_vl_moe.py +126 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/model/smollm3.py +17 -7
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model/smolvlm.py +158 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/monkey_patch.py +855 -52
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/multi_token_attention.py +1 -1
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/poly_norm.py +42 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/qwen2vl_mrope.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/rms_norm.py +15 -3
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/rope.py +64 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/softmax.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/sparsemax.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/swiglu.py +18 -1
- liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/tiled_mlp.py +125 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/tvd.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/utils.py +52 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel_nightly.egg-info/PKG-INFO +13 -3
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel_nightly.egg-info/SOURCES.txt +34 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel_nightly.egg-info/requires.txt +1 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_cosine_loss.py +1 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_grpo_loss.py +224 -47
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_jsd_loss.py +27 -9
- liger_kernel_nightly-0.6.4.dev20260121000828/test/conftest.py +15 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/bf16/test_mini_models.py +615 -8
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/bf16/test_mini_models_multimodal.py +318 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/bf16/test_mini_models_with_logits.py +545 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/fp32/test_mini_models.py +585 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/fp32/test_mini_models_multimodal.py +480 -9
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/convergence/fp32/test_mini_models_with_logits.py +549 -1
- liger_kernel_nightly-0.6.4.dev20260121000828/test/resources/fake_configs/HuggingFaceTB/SmolVLM2-256M-Video-Instruct/tokenizer_config.json +1192 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/test/resources/fake_configs/Qwen/Qwen3-VL-4B-Instruct/tokenizer_config.json +63 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_auto_model.py +35 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_cross_entropy.py +126 -6
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_fused_linear_cross_entropy.py +341 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_geglu.py +4 -4
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_group_norm.py +5 -11
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_jsd.py +29 -1
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_layer_norm.py +1 -0
- liger_kernel_nightly-0.6.4.dev20260121000828/test/transformers/test_llama4_rope.py +149 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_monkey_patch.py +1086 -78
- liger_kernel_nightly-0.6.4.dev20260121000828/test/transformers/test_poly_norm.py +281 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_rms_norm.py +142 -16
- liger_kernel_nightly-0.6.4.dev20260121000828/test/transformers/test_tiled_mlp.py +197 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/utils.py +161 -8
- liger_kernel_nightly-0.6.2.dev20251005233549/.github/workflows/docs.yml +0 -32
- liger_kernel_nightly-0.6.2.dev20251005233549/src/liger_kernel/transformers/rope.py +0 -20
- liger_kernel_nightly-0.6.2.dev20251005233549/test/conftest.py +0 -8
- liger_kernel_nightly-0.6.2.dev20251005233549/test/convergence/fp32/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/ISSUE_TEMPLATE/bug_report.yaml +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/ISSUE_TEMPLATE/feature_request.yaml +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.github/pull_request_template.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/.gitignore +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/LICENSE +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/Makefile +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/NOTICE +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/README.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/benchmarks_visualizer.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_cpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_distill_cosine_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_distill_jsd_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_dpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_dyt.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_embedding.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_fused_add_rms_norm.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_fused_neighborhood_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_geglu.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_group_norm.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_grpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_jsd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_kl_div.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_kto_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_layer_norm.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_llama4_rope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_multi_token_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_orpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_rms_norm.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_rope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_simpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_softmax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_sparse_multi_token_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_sparsemax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/benchmark_swiglu.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/benchmark/scripts/utils.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/dev/fmt-requirements.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/dev/modal/benchmarks.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/Examples.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/Getting-Started.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/High-Level-APIs.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/Low-Level-APIs.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/banner.GIF +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/compose.gif +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/e2e-memory.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/e2e-tps.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/logo-banner.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/patch.gif +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/images/post-training.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/index.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/docs/license.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/alignment/accelerate_config.yaml +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/alignment/run_orpo.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/README.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/callback.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/config/fsdp_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/gemma_7b_mem.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/gemma_7b_tp.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/llama_mem_alloc.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/llama_tps.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/qwen_mem_alloc.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/img/qwen_tps.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/launch_on_modal.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/requirements.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/run_benchmarks.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/run_gemma.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/run_llama.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/run_qwen.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/run_qwen2_vl.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/training.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/huggingface/training_multimodal.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/lightning/README.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/lightning/requirements.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/lightning/training.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/README.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/callback.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Memory_Stage1_num_head_3.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Memory_Stage1_num_head_5.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Memory_Stage2_num_head_3.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Memory_Stage2_num_head_5.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Throughput_Stage1_num_head_3.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Throughput_Stage1_num_head_5.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Throughput_Stage2_num_head_3.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/docs/images/Throughput_Stage2_num_head_5.png +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/fsdp/acc-fsdp.conf +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/medusa_util.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/requirements.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/scripts/llama3_8b_medusa.sh +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/examples/medusa/train.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/licenses/LICENSE-Apache-2.0 +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/licenses/LICENSE-MIT-AutoAWQ +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/licenses/LICENSE-MIT-Efficient-Cross-Entropy +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/licenses/LICENSE-MIT-llmc +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/licenses/LICENSE-MIT-triton +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/mkdocs.yml +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/setup.cfg +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/README.md +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/cpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/dpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/functional.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/fused_linear_preference.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/fused_linear_unpaired_preference.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/kto_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/orpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/chunked_loss/simpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/env_report.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/experimental/embedding.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/experimental/mm_int8int2.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/fused_neighborhood_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/jsd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/llama4_rope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/multi_token_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/rope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/softmax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/sparsemax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/swiglu.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/ops/tvd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/experimental/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/fsdp.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/src/liger_kernel/ops → liger_kernel_nightly-0.6.4.dev20260121000828/src/liger_kernel/transformers/model}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/trainer/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/trainer/orpo_trainer.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/transformers/trainer_integration.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/triton/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel/triton/monkey_patch.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel_nightly.egg-info/dependency_links.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/src/liger_kernel_nightly.egg-info/top_level.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/src/liger_kernel/transformers/model → liger_kernel_nightly-0.6.4.dev20260121000828/test}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/test → liger_kernel_nightly-0.6.4.dev20260121000828/test/chunked_loss}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_cpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_dpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_kto_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_orpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/chunked_loss/test_simpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/test/chunked_loss → liger_kernel_nightly-0.6.4.dev20260121000828/test/convergence}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/test/convergence → liger_kernel_nightly-0.6.4.dev20260121000828/test/convergence/bf16}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549/test/convergence/bf16 → liger_kernel_nightly-0.6.4.dev20260121000828/test/convergence/fp32}/__init__.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Google/Gemma3/gemma-3-4b-it/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Google/Paligemma/paligemma-3b-pt-224/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Llava/llava-1.5-7b-hf/preprocessor_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Llava/llava-1.5-7b-hf/processor_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Llava/llava-1.5-7b-hf/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/OpenGVLab/InternVL3-1B-hf/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Qwen/Qwen2-VL-7B-Instruct/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/Qwen/Qwen2.5-VL-7B-Instruct/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/meta-llama/Llama-3.2-11B-Vision-Instruct/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/fake_configs/meta-llama/Llama-4-Scout-17B-16E-Instruct/tokenizer_config.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/scripts/generate_tokenized_dataset.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/tiny_shakespeare.txt +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/tiny_shakespeare_tokenized/data-00000-of-00001.arrow +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/tiny_shakespeare_tokenized/dataset_info.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/resources/tiny_shakespeare_tokenized/state.json +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_dyt.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_embedding.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_flex_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_fused_add_rms_norm.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_fused_neighborhood_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_grpo_loss.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_kl_div.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_mm_int8int2.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_multi_token_attention.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_rope.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_softmax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_sparsemax.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_swiglu.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_trainer_integration.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_transformers.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/transformers/test_tvd.py +0 -0
- {liger_kernel_nightly-0.6.2.dev20251005233549 → liger_kernel_nightly-0.6.4.dev20260121000828}/test/triton/test_triton_monkey_patch.py +0 -0
|
@@ -26,10 +26,10 @@ jobs:
|
|
|
26
26
|
|
|
27
27
|
steps:
|
|
28
28
|
- name: Checkout code
|
|
29
|
-
uses: actions/checkout@
|
|
29
|
+
uses: actions/checkout@v6
|
|
30
30
|
|
|
31
31
|
- name: Set up Python
|
|
32
|
-
uses: actions/setup-python@
|
|
32
|
+
uses: actions/setup-python@v6
|
|
33
33
|
with:
|
|
34
34
|
python-version: '3.10'
|
|
35
35
|
|
|
@@ -50,10 +50,10 @@ jobs:
|
|
|
50
50
|
|
|
51
51
|
steps:
|
|
52
52
|
- name: Checkout code
|
|
53
|
-
uses: actions/checkout@
|
|
53
|
+
uses: actions/checkout@v6
|
|
54
54
|
|
|
55
55
|
- name: Set up Python
|
|
56
|
-
uses: actions/setup-python@
|
|
56
|
+
uses: actions/setup-python@v6
|
|
57
57
|
with:
|
|
58
58
|
python-version: '3.10'
|
|
59
59
|
|
|
@@ -35,22 +35,26 @@ jobs:
|
|
|
35
35
|
OUTPUT_DIR: benchmarks
|
|
36
36
|
OUTPUT_FILENAME: benchmark.csv
|
|
37
37
|
GENERATED_CSV: benchmark/data/all_benchmark_data.csv
|
|
38
|
+
# Sanitize user-controlled inputs by declaring them as environment variables
|
|
39
|
+
# This prevents command injection attacks by filtering dangerous characters
|
|
40
|
+
INPUT_COMMIT_HASH: ${{ github.event.inputs.commit_hash }}
|
|
41
|
+
INPUT_OVERWRITE: ${{ github.event.inputs.overwrite }}
|
|
38
42
|
|
|
39
43
|
|
|
40
44
|
steps:
|
|
41
45
|
# Step: Decide the commit hash to use
|
|
42
46
|
# Step: Checkout full history so we can check out any commit
|
|
43
47
|
- name: Checkout full repo history
|
|
44
|
-
uses: actions/checkout@
|
|
48
|
+
uses: actions/checkout@v6
|
|
45
49
|
with:
|
|
46
50
|
fetch-depth: 0 # Important: so we can checkout arbitrary commit
|
|
47
51
|
|
|
48
52
|
- name: Determine commit hash to checkout
|
|
49
53
|
id: choose_commit
|
|
50
54
|
run: |
|
|
51
|
-
if [ "${{ github.event_name}}" == "workflow_dispatch" ] && [ "$
|
|
52
|
-
echo "Using manual input commit: $
|
|
53
|
-
echo "hash=$
|
|
55
|
+
if [ "${{ github.event_name}}" == "workflow_dispatch" ] && [ "$INPUT_COMMIT_HASH" != "main" ]; then
|
|
56
|
+
echo "Using manual input commit: $INPUT_COMMIT_HASH"
|
|
57
|
+
echo "hash=$INPUT_COMMIT_HASH" >> $GITHUB_OUTPUT
|
|
54
58
|
else
|
|
55
59
|
echo "Using latest commit from main"
|
|
56
60
|
echo "hash=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
|
|
@@ -60,10 +64,10 @@ jobs:
|
|
|
60
64
|
- name: Replace benchmark folder from main (manual only, commit ≠ main)
|
|
61
65
|
if: ${{ github.event_name == 'workflow_dispatch' && github.event.inputs.commit_hash != 'main' }}
|
|
62
66
|
run: |
|
|
63
|
-
echo "Detected manual trigger with commit_hash = $
|
|
67
|
+
echo "Detected manual trigger with commit_hash = $INPUT_COMMIT_HASH"
|
|
64
68
|
|
|
65
69
|
# Save current branch (detached HEAD at old commit)
|
|
66
|
-
ORIG_COMMIT
|
|
70
|
+
ORIG_COMMIT="$INPUT_COMMIT_HASH"
|
|
67
71
|
|
|
68
72
|
# Fetch and checkout main
|
|
69
73
|
git fetch origin main
|
|
@@ -72,7 +76,7 @@ jobs:
|
|
|
72
76
|
# Save benchmark folder from main
|
|
73
77
|
cp -r benchmark /tmp/benchmark_main
|
|
74
78
|
# Checkout back to target commit
|
|
75
|
-
git checkout $ORIG_COMMIT
|
|
79
|
+
git checkout "$ORIG_COMMIT"
|
|
76
80
|
# Replace old benchmark with one from main
|
|
77
81
|
rm -rf benchmark
|
|
78
82
|
cp -r /tmp/benchmark_main benchmark
|
|
@@ -85,7 +89,7 @@ jobs:
|
|
|
85
89
|
|
|
86
90
|
if curl --output /dev/null --silent --head --fail "$BENCHMARK_URL"; then
|
|
87
91
|
echo "Benchmark already exists for commit $COMMIT_HASH"
|
|
88
|
-
if [ "$
|
|
92
|
+
if [ "$INPUT_OVERWRITE" != "true" ]; then
|
|
89
93
|
echo "Overwrite is false - exiting"
|
|
90
94
|
exit 1
|
|
91
95
|
else
|
|
@@ -96,7 +100,7 @@ jobs:
|
|
|
96
100
|
fi
|
|
97
101
|
|
|
98
102
|
- name: Set up Python
|
|
99
|
-
uses: actions/setup-python@
|
|
103
|
+
uses: actions/setup-python@v6
|
|
100
104
|
with:
|
|
101
105
|
python-version: '3.10'
|
|
102
106
|
|
|
@@ -117,7 +121,7 @@ jobs:
|
|
|
117
121
|
|
|
118
122
|
# Step 5: Checkout gh-pages branch in a subfolderAdd commentMore actions
|
|
119
123
|
- name: Checkout gh-pages
|
|
120
|
-
uses: actions/checkout@
|
|
124
|
+
uses: actions/checkout@v6
|
|
121
125
|
with:
|
|
122
126
|
ref: gh-pages
|
|
123
127
|
path: gh-pages
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
name: Publish documentation
|
|
2
|
+
on:
|
|
3
|
+
push:
|
|
4
|
+
branches:
|
|
5
|
+
- main
|
|
6
|
+
paths:
|
|
7
|
+
- 'docs/**'
|
|
8
|
+
- 'mkdocs.yml'
|
|
9
|
+
|
|
10
|
+
permissions:
|
|
11
|
+
contents: write
|
|
12
|
+
jobs:
|
|
13
|
+
deploy:
|
|
14
|
+
runs-on: ubuntu-latest
|
|
15
|
+
steps:
|
|
16
|
+
- uses: actions/checkout@v6
|
|
17
|
+
- name: Configure Git Credentials
|
|
18
|
+
run: |
|
|
19
|
+
git config user.name github-actions[bot]
|
|
20
|
+
git config user.email 41898282+github-actions[bot]@users.noreply.github.com
|
|
21
|
+
- uses: actions/setup-python@v6
|
|
22
|
+
with:
|
|
23
|
+
python-version: 3.x
|
|
24
|
+
- run: echo "cache_id=$(date --utc '+%V')" >> $GITHUB_ENV
|
|
25
|
+
- uses: actions/cache@v5
|
|
26
|
+
with:
|
|
27
|
+
key: mkdocs-material-${{ env.cache_id }}
|
|
28
|
+
path: .cache
|
|
29
|
+
restore-keys: |
|
|
30
|
+
mkdocs-material-
|
|
31
|
+
- run: pip install mkdocs-material mkdocstrings[python]
|
|
32
|
+
# ====== Backup the benchmarks from gh-pages ======
|
|
33
|
+
# This is necessary because the benchmarks are not included in the documentation build process.
|
|
34
|
+
# So we need to backup the benchmarks from gh-pages and restore them after the documentation is built.
|
|
35
|
+
- name: Backup benchmarks from gh-pages
|
|
36
|
+
run: |
|
|
37
|
+
git fetch origin gh-pages
|
|
38
|
+
# create worktree bound to local gh-pages, tracking origin/gh-pages
|
|
39
|
+
git branch -f gh-pages origin/gh-pages || true
|
|
40
|
+
mkdir -p ghp && git worktree add ghp gh-pages || true
|
|
41
|
+
if [ -d ghp/benchmarks ]; then
|
|
42
|
+
tar -C ghp -czf /tmp/benchmarks.tgz benchmarks
|
|
43
|
+
fi
|
|
44
|
+
# IMPORTANT: remove worktree so gh-pages isn't checked out anywhere
|
|
45
|
+
git worktree remove ghp --force || true
|
|
46
|
+
echo "Backed up benchmarks from gh-pages"
|
|
47
|
+
# ====== Deploy the documentation ======
|
|
48
|
+
- name: Deploy documentation
|
|
49
|
+
run: mkdocs gh-deploy --force
|
|
50
|
+
# ====== Restore the benchmarks onto gh-pages ======
|
|
51
|
+
# This is necessary because the benchmarks are not included in the documentation build process.
|
|
52
|
+
# So we need to restore the benchmarks onto gh-pages after the documentation is built.
|
|
53
|
+
- name: Restore benchmarks onto gh-pages
|
|
54
|
+
run: |
|
|
55
|
+
# Refresh remote tracking and recreate a clean worktree
|
|
56
|
+
git fetch origin gh-pages
|
|
57
|
+
git worktree add -B gh-pages ghp origin/gh-pages
|
|
58
|
+
if [ -f /tmp/benchmarks.tgz ]; then
|
|
59
|
+
tar -C ghp -xzf /tmp/benchmarks.tgz
|
|
60
|
+
git -C ghp add -A
|
|
61
|
+
git -C ghp commit -m "Restore benchmarks after gh-deploy" || echo "No changes"
|
|
62
|
+
git -C ghp push origin gh-pages
|
|
63
|
+
fi
|
|
64
|
+
git worktree remove ghp --force || true
|
|
@@ -26,10 +26,10 @@ jobs:
|
|
|
26
26
|
|
|
27
27
|
steps:
|
|
28
28
|
- name: Checkout code
|
|
29
|
-
uses: actions/checkout@
|
|
29
|
+
uses: actions/checkout@v6
|
|
30
30
|
|
|
31
31
|
- name: Set up Python
|
|
32
|
-
uses: actions/setup-python@
|
|
32
|
+
uses: actions/setup-python@v6
|
|
33
33
|
with:
|
|
34
34
|
python-version: '3.10'
|
|
35
35
|
|
|
@@ -58,7 +58,7 @@ jobs:
|
|
|
58
58
|
apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
59
59
|
|
|
60
60
|
- name: Checkout code
|
|
61
|
-
uses: actions/checkout@
|
|
61
|
+
uses: actions/checkout@v6
|
|
62
62
|
|
|
63
63
|
- name: Setup Dependencies
|
|
64
64
|
shell: bash
|
|
@@ -25,10 +25,10 @@ jobs:
|
|
|
25
25
|
|
|
26
26
|
steps:
|
|
27
27
|
- name: Checkout code
|
|
28
|
-
uses: actions/checkout@
|
|
28
|
+
uses: actions/checkout@v6
|
|
29
29
|
|
|
30
30
|
- name: Set up Python
|
|
31
|
-
uses: actions/setup-python@
|
|
31
|
+
uses: actions/setup-python@v6
|
|
32
32
|
with:
|
|
33
33
|
python-version: '3.10'
|
|
34
34
|
|
|
@@ -49,10 +49,10 @@ jobs:
|
|
|
49
49
|
|
|
50
50
|
steps:
|
|
51
51
|
- name: Checkout code
|
|
52
|
-
uses: actions/checkout@
|
|
52
|
+
uses: actions/checkout@v6
|
|
53
53
|
|
|
54
54
|
- name: Set up Python
|
|
55
|
-
uses: actions/setup-python@
|
|
55
|
+
uses: actions/setup-python@v6
|
|
56
56
|
with:
|
|
57
57
|
python-version: '3.10'
|
|
58
58
|
|
|
@@ -75,10 +75,10 @@ jobs:
|
|
|
75
75
|
|
|
76
76
|
steps:
|
|
77
77
|
- name: Checkout code
|
|
78
|
-
uses: actions/checkout@
|
|
78
|
+
uses: actions/checkout@v6
|
|
79
79
|
|
|
80
80
|
- name: Set up Python
|
|
81
|
-
uses: actions/setup-python@
|
|
81
|
+
uses: actions/setup-python@v6
|
|
82
82
|
with:
|
|
83
83
|
python-version: '3.10'
|
|
84
84
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: liger_kernel_nightly
|
|
3
|
-
Version: 0.6.
|
|
3
|
+
Version: 0.6.4.dev20260121000828
|
|
4
4
|
Summary: Efficient Triton kernels for LLM Training
|
|
5
5
|
License: BSD 2-CLAUSE LICENSE
|
|
6
6
|
Copyright 2024 LinkedIn Corporation
|
|
@@ -45,6 +45,7 @@ Requires-Dist: datasets>=2.19.2; extra == "dev"
|
|
|
45
45
|
Requires-Dist: seaborn; extra == "dev"
|
|
46
46
|
Requires-Dist: mkdocs-material; extra == "dev"
|
|
47
47
|
Requires-Dist: torchvision>=0.20; extra == "dev"
|
|
48
|
+
Requires-Dist: prek>=0.2.28; extra == "dev"
|
|
48
49
|
|
|
49
50
|
<a name="readme-top"></a>
|
|
50
51
|
|
|
@@ -79,8 +80,8 @@ Requires-Dist: torchvision>=0.20; extra == "dev"
|
|
|
79
80
|
</a>
|
|
80
81
|
</td>
|
|
81
82
|
<td style="padding: 10px;">
|
|
82
|
-
<a href="https://discord.gg/
|
|
83
|
-
<img src="https://dcbadge.limes.pink/api/server/
|
|
83
|
+
<a href="https://discord.gg/X4MaxPgA">
|
|
84
|
+
<img src="https://dcbadge.limes.pink/api/server/https://discord.gg/X4MaxPgA?style=flat" alt="Join Our Discord">
|
|
84
85
|
</a>
|
|
85
86
|
</td>
|
|
86
87
|
</tr>
|
|
@@ -95,6 +96,7 @@ Requires-Dist: torchvision>=0.20; extra == "dev"
|
|
|
95
96
|
<details>
|
|
96
97
|
<summary>Latest News 🔥</summary>
|
|
97
98
|
|
|
99
|
+
- [2025/12/19] We announced a liger kernel discord channel at https://discord.gg/X4MaxPgA; We will be hosting Liger Kernel x Triton China Meetup in mid of January 2026
|
|
98
100
|
- [2025/03/06] We release a joint blog post on TorchTune × Liger - [Peak Performance, Minimized Memory: Optimizing torchtune’s performance with torch.compile & Liger Kernel](https://pytorch.org/blog/peak-performance-minimized-memory/)
|
|
99
101
|
- [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)!
|
|
100
102
|
- [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)
|
|
@@ -113,6 +115,8 @@ We've also added optimized Post-Training kernels that deliver **up to 80% memory
|
|
|
113
115
|
|
|
114
116
|
You can view the documentation site for additional installation, usage examples, and API references:https://linkedin.github.io/Liger-Kernel/
|
|
115
117
|
|
|
118
|
+
You can view the Liger Kernel Technical Report: https://openreview.net/forum?id=36SjAIT42G
|
|
119
|
+
|
|
116
120
|
## Supercharge Your Model with Liger Kernel
|
|
117
121
|
|
|
118
122
|

|
|
@@ -310,8 +314,12 @@ loss.backward()
|
|
|
310
314
|
| Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
311
315
|
| Granite 3.0 & 3.1 | `liger_kernel.transformers.apply_liger_kernel_to_granite` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss |
|
|
312
316
|
| OLMo2 | `liger_kernel.transformers.apply_liger_kernel_to_olmo2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
317
|
+
| Olmo3 | `liger_kernel.transformers.apply_liger_kernel_to_olmo3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
313
318
|
| GLM-4 | `liger_kernel.transformers.apply_liger_kernel_to_glm4` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
319
|
+
| GPT-OSS | `liger_kernel.transformers.apply_liger_kernel_to_gpt_oss` | RoPE, RMSNorm, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
314
320
|
| InternVL3 | `liger_kernel.transformers.apply_liger_kernel_to_internvl` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
321
|
+
| HunyuanV1 | `liger_kernel.transformers.apply_liger_kernel_to_hunyuan_v1_dense` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
322
|
+
| HunyuanV1 MoE | `liger_kernel.transformers.apply_liger_kernel_to_hunyuan_v1_moe` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
315
323
|
|
|
316
324
|
|
|
317
325
|
## Low-level APIs
|
|
@@ -438,3 +446,5 @@ url={https://openreview.net/forum?id=36SjAIT42G}
|
|
|
438
446
|
↑ Back to Top ↑
|
|
439
447
|
</a>
|
|
440
448
|
</p>
|
|
449
|
+
|
|
450
|
+
|
|
@@ -31,8 +31,8 @@
|
|
|
31
31
|
</a>
|
|
32
32
|
</td>
|
|
33
33
|
<td style="padding: 10px;">
|
|
34
|
-
<a href="https://discord.gg/
|
|
35
|
-
<img src="https://dcbadge.limes.pink/api/server/
|
|
34
|
+
<a href="https://discord.gg/X4MaxPgA">
|
|
35
|
+
<img src="https://dcbadge.limes.pink/api/server/https://discord.gg/X4MaxPgA?style=flat" alt="Join Our Discord">
|
|
36
36
|
</a>
|
|
37
37
|
</td>
|
|
38
38
|
</tr>
|
|
@@ -47,6 +47,7 @@
|
|
|
47
47
|
<details>
|
|
48
48
|
<summary>Latest News 🔥</summary>
|
|
49
49
|
|
|
50
|
+
- [2025/12/19] We announced a liger kernel discord channel at https://discord.gg/X4MaxPgA; We will be hosting Liger Kernel x Triton China Meetup in mid of January 2026
|
|
50
51
|
- [2025/03/06] We release a joint blog post on TorchTune × Liger - [Peak Performance, Minimized Memory: Optimizing torchtune’s performance with torch.compile & Liger Kernel](https://pytorch.org/blog/peak-performance-minimized-memory/)
|
|
51
52
|
- [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)!
|
|
52
53
|
- [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,6 +66,8 @@ We've also added optimized Post-Training kernels that deliver **up to 80% memory
|
|
|
65
66
|
|
|
66
67
|
You can view the documentation site for additional installation, usage examples, and API references:https://linkedin.github.io/Liger-Kernel/
|
|
67
68
|
|
|
69
|
+
You can view the Liger Kernel Technical Report: https://openreview.net/forum?id=36SjAIT42G
|
|
70
|
+
|
|
68
71
|
## Supercharge Your Model with Liger Kernel
|
|
69
72
|
|
|
70
73
|

|
|
@@ -262,8 +265,12 @@ loss.backward()
|
|
|
262
265
|
| Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
263
266
|
| Granite 3.0 & 3.1 | `liger_kernel.transformers.apply_liger_kernel_to_granite` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss |
|
|
264
267
|
| OLMo2 | `liger_kernel.transformers.apply_liger_kernel_to_olmo2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
268
|
+
| Olmo3 | `liger_kernel.transformers.apply_liger_kernel_to_olmo3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
265
269
|
| GLM-4 | `liger_kernel.transformers.apply_liger_kernel_to_glm4` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
270
|
+
| GPT-OSS | `liger_kernel.transformers.apply_liger_kernel_to_gpt_oss` | RoPE, RMSNorm, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
266
271
|
| InternVL3 | `liger_kernel.transformers.apply_liger_kernel_to_internvl` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
272
|
+
| HunyuanV1 | `liger_kernel.transformers.apply_liger_kernel_to_hunyuan_v1_dense` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
273
|
+
| HunyuanV1 MoE | `liger_kernel.transformers.apply_liger_kernel_to_hunyuan_v1_moe` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy |
|
|
267
274
|
|
|
268
275
|
|
|
269
276
|
## Low-level APIs
|
|
@@ -390,3 +397,5 @@ url={https://openreview.net/forum?id=36SjAIT42G}
|
|
|
390
397
|
↑ Back to Top ↑
|
|
391
398
|
</a>
|
|
392
399
|
</p>
|
|
400
|
+
|
|
401
|
+
|