← Search

Zhongkai Zhao

3 accepted papers

2026

OmniScale: Scaling Any Modality Model Training with Model-Centric Distributed Recipe Zoo

AAAI 2026technical

Recent advances in large language models (LLMs) have driven impressive progress in omni-modal understanding and generation. However, training omni-modal LLMs remains a significant challenge due to the heterogeneous model architectures required to process diverse modalities, necessitating sophisticat

Cited by 0SourcePDFScholar
2025

StarTrail: Concentric Ring Sequence Parallelism for Efficient Near-Infinite-Context Transformer Model Training

NeurIPS 2025poster

Training Transformer models on long sequences in a distributed setting poses significant challenges in terms of efficiency and scalability. Current methods are either constrained by the number of attention heads or excessive communication overheads. To address this problem, we propose StarTrail, a m…

Cited by 0SourceScholar
2024

What If the TV Was Off? Examining Counterfactual Reasoning Abilities of Multi-modal Language Models

CVPR 2024poster

Counterfactual reasoning a fundamental aspect of human cognition involves contemplating alternatives to established facts or past events significantly enhancing our abilities in planning and decision-making. In light of the advancements in current multi-modal large language models we explore their e…