New Kernels Ensure Deterministic LLM Inference Across Tensor Parallel Sizes

Researchers have developed Tree-Based Invariant Kernels (TBIK) to ensure deterministic inference in large language models, regardless of tensor parallel (TP) size. This addresses a critical issue where identical inputs can produce different outputs due to variations in TP size and floating-point arithmetic. TBIK guarantees bit-wise reproducibility by aligning reduction orders through a hierarchical binary tree structure, which is crucial for applications like LLM-as-a-judge and reinforcement learning. AI

IMPACT Ensures consistent LLM outputs for critical applications like RL and evaluation, removing a key barrier to reliable deployment.

RANK_REASON The cluster contains an academic paper detailing a new technical method for improving LLM inference. [lever_c_demoted from research: ic=1 ai=1.0]

Read on arXiv stat.ML →

• large language models
• tensor parallel
• Tree-Based Invariant Kernels
• vLLM
• Xinheng Ding

• paper
• infra

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →