Author

Jiajun Wu

3 papers indexed here

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 May 2026

Anon: Extrapolating Adaptivity Beyond SGD and Adam

Anon is proposed, an optimizer that achieves fully continuous adaptivity extrapolation across the entire real-number spectrum and demonstrates highly competitive and scalable performance among state-of-the-art element-wise optimizers on representative image classification, diffusion, and large language modeling tasks.

Yi-Heng Zhang, Kai-Yan Zhao, Shaowu Wu et al. · arXiv.org · 0 citations

#artificial intelligence Feb 2026

Shiva-DiT: Residual-Based Differentiable Top-k Selection for Efficient Diffusion Transformers

Shiva-DiT, based on Residual-Based Differentiable Top-k Selection, provides four fidelity-latency operating points and reaches a 1.54x wall-clock speedup with competitive fidelity.

Jiaji Zhang, Hai-Liang Zhao, Guoxuan Zhu et al. · arXiv.org · 2 citations

Preprint Aug 2026

Unified Lookup-Table Inference with Signed-Digit K/V Caches for Ternary LLMs

This work develops a unified lookup-table inference approach for ternary LLMs that stores runtime K/V states as scaled multi-plane signed digits organized around the reduction structure of attention.

Ziang Duan, Jia-Jun Wu, Ze-Tian Chen et al. · 0 citations