Types & classes — github.com/OpenNLPLab/lightning-attention

CodeHub MCPcopy Create free account

hub / github.com/OpenNLPLab/lightning-attention / types & classes

Types & classes7 in github.com/OpenNLPLab/lightning-attention

⨍ Functions 53 ◇ Types & classes 7 ↳ Endpoints 3

↓ 3 callersClassSimpleRMSNorm

lightning_attn/ops/triton/srmsnorm.py:178

↓ 3 callersClassSimpleRMSNormTorch

lightning_attn/ops/torch/srmsnorm.py:9

↓ 1 callersClassTnlAttention

examples/module/tnl_attn.py:12

ClassLightningAttention2

lightning_attn/ops/triton/lightning_attn2.py:398

ClassLightningAttention2NoDecay

lightning_attn/ops/triton/lightning_attn2_no_decay.py:367

ClassLightningAttention2Parallel

lightning_attn/ops/triton/lightning_attn2_parallel.py:835

lightning_attn/ops/triton/srmsnorm.py:76