Kang, Shijia

3 publications

NeurIPS 2025 Beyond Single-Task: Robust Multi-Task Length Generalization for LLMs Yi Hu, Shijia Kang, Haotong Yang, Haotian Xu, Muhan Zhang

ICLR 2025 Number Cookbook: Number Understanding of Language Models and How to Improve It Haotong Yang, Yi Hu, Shijia Kang, Zhouchen Lin, Muhan Zhang

ICLR 2025 On the Completeness of Invariant Geometric Deep Learning Models Zian Li, Xiyuan Wang, Shijia Kang, Muhan Zhang