Kernels:

---

kernels-staging

/

fla

Like 0

Follow

Kernels Staging 6

Trusted publisher

library_name: kernels
license: mit

This is the repository card of kernels-community/fla 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/fla", version=1)
chunk_kda = kernel_module.chunk_kda

chunk_kda(...)

Available functions

• chunk_kda
• fused_recurrent_kda
• chunk_gated_delta_rule
• fused_recurrent_gated_delta_rule

Available layers

• FusedRMSNormGated
• chunk_kimi_delta_attention
• recurrent_kimi_delta_attention
• chunk_gated_delta_rule
• recurrent_gated_delta_rule

Benchmarks

No benchmark available yet.