← Search

Sijun Zhang

3 accepted papers

2026

WeDLM: Reconciling Diffusion Language Models with Standard Causal Attention for Fast Inference

ICML 2026oral

Autoregressive (AR) generation is the standard decoding paradigm for Large Language Models (LLMs), but its token-by-token nature limits parallelism at inference time. Diffusion Language Models (DLLMs) offer parallel decoding by recovering multiple masked tokens per step; however, in practice they of…

Cited by 0SourceScholar
2025

HybridNorm: Towards Stable and Efficient Transformer Training via Hybrid Normalization

NeurIPS 2025poster

Transformers have become the de facto architecture for a wide range of machine learning tasks, particularly in large language models (LLMs). Despite their remarkable performance, many challenges remain in training deep transformer networks, especially regarding the position of the layer normalizatio…

Cited by 0SourcecodeScholar
2025

TC-MoE: Augmenting Mixture of Experts with Ternary Expert Choice

ICLR 2025poster

The Mixture of Experts (MoE) architecture has emerged as a promising solution to reduce computational overhead by selectively activating subsets of model parameters. The effectiveness of MoE models depends primarily on their routing mechanisms, with the widely adopted Top-K routing scheme used for a…