3 papers
cs.LG2026
Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization
Mengfan Liu, Da Zheng, Junwei Su +1
Despite the strong reasoning capabilities of large language models (LLMs), optimizing the execution efficiency of tensor programs remains challenging due to the need for precise, c…
cs.LG2026
Full-Graph vs. Mini-Batch Training: Comprehensive Analysis from a Batch Size and Fan-Out Size Perspective
Mengfan Liu, Da Zheng, Junwei Su +1
Full-graph and mini-batch Graph Neural Network (GNN) training approaches have distinct system design demands, making it crucial to choose the appropriate approach to develop. A cor…
cs.LG2025
Knowledge Augmented Complex Problem Solving with Large Language Models: A Survey
Da Zheng, Lun Du, Junwei Su +6
Problem-solving has been a fundamental driver of human progress in numerous domains. With advancements in artificial intelligence, Large Language Models (LLMs) have emerged as powe…