Walid Ahmed
2026
FlowHN: Adaptive Token Routing for Efficient Parallel Hybrid Networks
Mohammad Mahdi Moradi | Walid Ahmed | Shuangyue Wen | Sudhir Mudur | Weiwei Zhang | Yang Liu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
Mohammad Mahdi Moradi | Walid Ahmed | Shuangyue Wen | Sudhir Mudur | Weiwei Zhang | Yang Liu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
Production LLMs must balance modeling quality with predictable latency, stable accelerator utilization, and cost-efficient scaling—constraints that remain difficult for existing architectures. Transformers provide strong reasoning but incur quadratic complexity, while state-space models (SSMs) scale efficiently yet lack fine-grained interactions; prior hybrids either introduce sequential bottlenecks or rely on learned routing that complicates deployment. We present FlowHN, a deployment-oriented parallel hybrid architecture that enables deterministic conditional computation via FLOP-aware token circulation across attention and SSM branches. Instead of dynamic expert routing, FlowHN performs hardware-aligned token scheduling that balances workloads, reduces synchronization stalls, and preserves full parameter utilization. Across 135M–1B models, FlowHN achieves up to 4× higher throughput and 15% higher MFU than strong Transformer, SSM, and hybrid baselines while maintaining competitive accuracy on reasoning, coding, and long-context tasks up to 32K tokens. FlowHN is designed to integrate directly into existing Hybrid pipelines without changes to optimizers, training stacks, or inference serving infrastructure, making it practical for real-world deployment.
2025
ECHO-LLaMA: Efficient Caching for High-Performance LLaMA Training
Maryam Dialameh | Rezaul Karim | Hossein Rajabzadeh | Omar Mohamed Awad | Boxing Chen | Hyock Ju Kwon | Walid Ahmed | Yang Liu
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track
Maryam Dialameh | Rezaul Karim | Hossein Rajabzadeh | Omar Mohamed Awad | Boxing Chen | Hyock Ju Kwon | Walid Ahmed | Yang Liu
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track
This paper introduces ECHO-LLaMA, an efficient LLaMA architecture designed to improve both the training speed and inference throughput of LLaMA architectures while maintaining its learning capacity. ECHO-LLaMA transforms LLaMA models into shared KV caching across certain layers, significantly reducing KV computational complexity while maintaining or improving language performance. Experimental results demonstrate that ECHO-LLaMA achieves up to 77% higher token-per-second throughput during training, up to 16% higher Model FLOPs Utilization (MFU), and up to 14% lower loss when trained on an equal number of tokens. Furthermore, on the 1.1B model, ECHO-LLaMA delivers approximately 7% higher test-time throughput compared to the baseline. By introducing a computationally efficient adaptation mechanism, ECHO-LLaMA offers a scalable and cost-effective solution for pretraining and finetuning large language models, enabling faster and more resource-efficient training without compromising performance.