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.
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/PKG-INFO +6 -6
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/README.md +5 -5
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_cpo_loss.py +8 -15
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_dpo_loss.py +4 -4
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_orpo_loss.py +8 -15
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_simpo_loss.py +8 -15
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/pyproject.toml +1 -1
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/PKG-INFO +6 -6
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/ISSUE_TEMPLATE/bug_report.yaml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/ISSUE_TEMPLATE/feature_request.yaml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/pull_request_template.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/amd-ci.yml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/nvi-ci.yml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/publish-nightly.yml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.github/workflows/publish-release.yml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/.gitignore +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/LICENSE +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/Makefile +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/NOTICE +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/benchmarks_visualizer.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/data/all_benchmark_data.csv +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_embedding.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_fused_linear_cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_geglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_group_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_kl_div.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_layer_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_rms_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_rope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/benchmark_swiglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/benchmark/scripts/utils.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/fmt-requirements.txt +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/modal/tests.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/dev/modal/tests_bwd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/Acknowledgement.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/CONTRIBUTING.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/License.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/banner.GIF +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/compose.gif +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/e2e-memory.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/e2e-tps.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/logo-banner.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/patch.gif +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/docs/images/post-training.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/alignment/accelerate_config.yaml +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/alignment/run_orpo.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/README.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/callback.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/config/fsdp_config.json +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/gemma_7b_mem.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/gemma_7b_tp.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/llama_mem_alloc.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/llama_tps.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/qwen_mem_alloc.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/img/qwen_tps.png +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/launch_on_modal.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/requirements.txt +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_benchmarks.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_gemma.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_llama.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_qwen.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/run_qwen2_vl.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/training.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/huggingface/training_multimodal.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/README.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/requirements.txt +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/lightning/training.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/README.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/callback.py +0 -0
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/fsdp/acc-fsdp.conf +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/medusa_util.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/requirements.txt +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/scripts/llama3_8b_medusa.sh +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/examples/medusa/train.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-Apache-2.0 +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-AutoAWQ +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-Efficient-Cross-Entropy +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-llmc +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/licenses/LICENSE-MIT-triton +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/setup.cfg +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/setup.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/README.md +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/cpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/dpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/functional.py +0 -0
- {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
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/orpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/chunked_loss/simpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/env_report.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/experimental/embedding.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/experimental/mm_int8int2.py +0 -0
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/geglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/group_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/kl_div.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/layer_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/rms_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/rope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/swiglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/ops/utils.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/auto_model.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/experimental/embedding.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/functional.py +0 -0
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/geglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/group_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/kl_div.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/layer_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/gemma.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/gemma2.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/llama.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mistral.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mixtral.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/mllama.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/phi3.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/qwen2.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/model/qwen2_vl.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/monkey_patch.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/rms_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/rope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/swiglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer/orpo_trainer.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/transformers/trainer_integration.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/triton/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/triton/monkey_patch.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel/utils.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/SOURCES.txt +0 -0
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/src/liger_kernel_nightly.egg-info/requires.txt +0 -0
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_cpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_dpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_orpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/chunked_loss/test_simpo_loss.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/conftest.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/__init__.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models_multimodal.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/convergence/test_mini_models_with_logits.py +0 -0
- {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
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/scripts/generate_tokenized_dataset.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare.txt +0 -0
- {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
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare_tokenized/dataset_info.json +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/resources/tiny_shakespeare_tokenized/state.json +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_auto_model.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_embedding.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_fused_linear_cross_entropy.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_fused_linear_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_geglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_group_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_jsd.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_kl_div.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_layer_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_mm_int8int2.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_monkey_patch.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_qwen2vl_mrope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_rms_norm.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_rope.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_swiglu.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_trainer_integration.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/transformers/test_transformers.py +0 -0
- {liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/test/triton/test_triton_monkey_patch.py +0 -0
- {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.
|
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
|
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
|
-
|
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
|
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
|
-
|
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
|
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
|
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
|
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
|
-
|
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
|
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
|
-
|
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
|
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
|
-
|
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
|
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
|
-
|
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.
|
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.
|
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.
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
{liger_kernel_nightly-0.5.2.dev20250109023714 → liger_kernel_nightly-0.5.2.dev20250115025212}/NOTICE
RENAMED
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|