Mojo module
fused_qk_rope
Functions
- fused_qk_rope:
- fused_qk_rope_ragged: Applies RoPE (Rotary Position Embedding) to query and key tensors.
- get_identity_rope_coeff:
- get_safetensors_idx:
- rope_k_cache:
- rope_q_proj:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!
