This is the repository card of kernels-community/liger-kernels that has been pushed on the Hub. It was built to be used with the kernels library. This card was automatically generated.

How to use

# make sure `kernels` is installed: `pip install -U kernels`
from kernels import get_kernel

kernel_module = get_kernel("kernels-community/liger-kernels", version=3)
LigerForCausalLMLoss = kernel_module.LigerForCausalLMLoss

LigerForCausalLMLoss(...)

Available functions

  • LigerForCausalLMLoss
  • liger_rotary_pos_emb

Available layers

  • LigerRMSNorm
  • LigerLinear
  • LigerSwiGLUMLP
  • LigerGEGLUMLP
  • LigerTiledSwiGLUMLP
  • LigerTiledGEGLUMLP
  • CrossEntropyOutput
  • liger_rotary_pos_emb_layer
  • LigerForCausalLMLossLayer

Benchmarks

No benchmark available yet.

Downloads last month
-
bsd-2-clause