/**
 * RoPE (Rotary Position Embedding) kernels
 */
export interface RoPEConfig {
    head_dim: number;
    rope_scale: number;
    rope_theta: number;
    dtype: 'f32' | 'f16';
}
export declare function generateRoPEShader(config: RoPEConfig): string;
export interface Llama31RoPEConfig {
    head_dim: number;
    rope_theta: number;
    low_freq_factor: number;
    high_freq_factor: number;
    old_context_len: number;
    dtype: 'f32' | 'f16';
}
export declare function generateLlama31RoPEShader(config: Llama31RoPEConfig): string;