3 papers
cs.DC2026
HetRL: Efficient Reinforcement Learning for LLMs in Heterogeneous Environments
Yongjun He, Shuai Zhang, Jiading Gai +5
As large language models (LLMs) continue to scale and new GPUs are released even more frequently, there is an increasing demand for LLM post-training in heterogeneous environments…
stat.ML2025
Variational Causal Inference
Yulun Wu, Layne C. Price, Zichen Wang +3
Estimating an individual's potential outcomes under counterfactual treatments is a challenging task for traditional causal inference and supervised learning approaches when the out…
cs.LG2025
Predicting Cellular Responses with Variational Causal Inference and Refined Relational Information
Yulun Wu, Robert A. Barton, Zichen Wang +5
Predicting the responses of a cell under perturbations may bring important benefits to drug discovery and personalized therapeutics. In this work, we propose a novel graph variatio…