/**
 * Sampling kernels for token generation
 */
export interface SamplingConfig {
    vocab_size: number;
    batch_size: number;
    dtype: 'f32' | 'f16';
}
export declare function generateCategoricalSamplingShader(config: SamplingConfig): string;
export interface TopKSamplingConfig {
    vocab_size: number;
    batch_size: number;
    top_k: number;
    dtype: 'f32' | 'f16';
}
export declare function generateTopKSamplingShader(config: TopKSamplingConfig): string;
export interface TopPSamplingConfig {
    vocab_size: number;
    batch_size: number;
    top_p: number;
    dtype: 'f32' | 'f16';
}
export declare function generateTopPSamplingShader(config: TopPSamplingConfig): string;