Skip to content

Author

Jian-Wei Zhang

We have 6 of 9 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#artificial intelligence Preprint Oct 2026

D2K-Bench: Can LLM Agents Turn Expert Designs into Efficient GPU Kernels?

GPU kernels generated by large language model (LLM) agents can remain less efficient than expert implementations, but runtime alone does not reveal how the gap relates to design discovery and implementation. We introduce D2K-Bench, a diagnostic benchmark of 26 tasks and 85 workloads that measures how effectively agents...

Dai-Feng Li, Huiqiang Jiang, Chengruidong Zhang et al. · 0 citations
#machine learning Preprint Sep 2026

QwenGyre: An Elastic Reinforcement Learning Framework for Training xLong-Horizon Agents

QwenGyre elastically reallocates GPUs between rollout and training without interrupting live executions, while its trajectory processor reconstructs branching histories, scores partial progress, and deduplicates redundant paths to bound training costs.

Wei-Qi Wang, Yu-Xin Zhou, Mou-Xiang Chen et al. · 0 citations
Jul 2026

QUADS: Stabilizing NVFP4 Reinforcement Learning for MoE via QUantization-error Alignment across Dual Sides

This work identifies activation error, rather than weight error, as the dominant source of FP4 RL instability: weights can be synchronized and aligned by a shared quantization-dequantization path, whereas activations are recomputed online and error is amplified by the coarse E2M1 grid.

Zhengyang Zhuge, Hao Yu, Xin Wang et al. · 0 citations
#artificial intelligence Conference Aug 2026

CateKV: On Sequential Consistency for Long-Context LLM Inference Acceleration

This work discovers that certain attention heads exhibit sequential consistency in their attention patterns, which can be persistently identified using a coefficient-of-variation-based algorithm, and proposes CateKV, a hybrid KV cache method that retains only critical token information for consistent heads, thereby red...

Hao-Yun Jiang, Hao-Lin Li, Jian-Wei Zhang et al. · 2 citations
#natural language process... Preprint Aug 2026

H-Scale: Hessian-Guided Scale Refinement for NVFP4 Sub-Byte LLM Inference

H-Scale is a lightweight post-processing method for NVFP4 per-group scale refinement that selects hardware-valid group scales using a diagonal second-order proxy derived from calibration activations, thereby targeting layer output perturbation more directly.

Hao Yu, Zheng Li, Dayiheng Liu et al. · 1 citation · ⚡1

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.