# High-Level APIs ## AutoModel | **AutoModel Variant** | **API** | |------------------------|---------| | AutoModelForCausalLM | `liger_kernel.transformers.AutoLigerKernelForCausalLM` | This API extends the implementation of the `AutoModelForCausalLM` within the `transformers` library from Hugging Face. ::: liger_kernel.transformers.AutoLigerKernelForCausalLM options: extra: show_docstring: true show_signature: true show_source: true !!! Example "Try it Out" You can experiment as shown in this example [here](https://github.com/linkedin/Liger-Kernel?tab=readme-ov-file#1-use-autoligerkernelforcausallm). --- ## Patching You can also use the Patching APIs to use the kernels for a specific model architecture. | **Model** | **API** | **Supported Operations** | |-------------|--------------------------------------------------------------|-------------------------------------------------------------------------| | LLaMA 2 & 3 | `liger_kernel.transformers.apply_liger_kernel_to_llama` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | LLaMA 3.2-Vision | `liger_kernel.transformers.apply_liger_kernel_to_mllama` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Mistral | `liger_kernel.transformers.apply_liger_kernel_to_mistral` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Mixtral | `liger_kernel.transformers.apply_liger_kernel_to_mixtral` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Gemma1 | `liger_kernel.transformers.apply_liger_kernel_to_gemma` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Gemma2 | `liger_kernel.transformers.apply_liger_kernel_to_gemma2` | RoPE, RMSNorm, GeGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Qwen2, Qwen2.5, & QwQ | `liger_kernel.transformers.apply_liger_kernel_to_qwen2` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Qwen2-VL | `liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl` | RMSNorm, LayerNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | | Phi3 & Phi3.5 | `liger_kernel.transformers.apply_liger_kernel_to_phi3` | RoPE, RMSNorm, SwiGLU, CrossEntropyLoss, FusedLinearCrossEntropy | ### Function Signatures ::: liger_kernel.transformers.apply_liger_kernel_to_llama options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_mllama options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_mistral options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_mixtral options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_gemma options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_gemma2 options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_qwen2 options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_qwen2_vl options: extra: show_docstring: true show_signature: true ::: liger_kernel.transformers.apply_liger_kernel_to_phi3 options: extra: show_docstring: true show_signature: true