Liger-Kernel
Fine-Tuning PlatformMLOps & Experiment Tracking

Liger-Kernel

Efficient Triton Kernels for LLM Training

Open Source

About

Liger Kernel is a collection of Triton kernels designed specifically for LLM training. It can effectively increase multi-GPU training throughput by 20% and reduces memory usage by 60%. We have implemented Hugging Face Compatible RMSNorm, RoPE, SwiGLU, CrossEntropy, FusedLinearCrossEntropy, and more to come. The kernel works out of the box with Flash Attention, PyTorch FSDP, and Microsoft DeepSpeed. We welcome contributions from the community to gather the best kernels for LLM training.

Open Source Health

Not enough history
Stars
6,610
Forks
599
License
BSD-2-Clause
Last commit
28 days ago
Python

Related Categories