Yulun "Rayn" Wu

yulun <dot> wu <at> capitalone.com

yulun <underscore> wu <at> berkeley.edu

google scholar github linkedin

I'm an applied researcher with the AI Foundations team at Capital One in the San Francisco Bay Area. My research interests include deep probabilistic modeling, causal inference, meta learning, and reinforcement learning. I earned my Ph.D. in biostatistics with D.E. in computational science and engineering at University of California, Berkeley.

Selected Publications

Zero-shot Meta-learning for Tabular Prediction Tasks with Adversarially Pre-trained Transformer
Yulun Wu, and Doron L. Bergman.
ICML 2025
paper / code

Counterfactual Generative Modeling with Variational Causal Inference
Yulun Wu, Louie McConnell, and Claudia Iriondo.
ICLR 2025
paper / code

Longitudinal Targeted Minimum Loss-based Estimation with Temporal-Difference Heterogeneous Transformer
Toru Shirakawa, Yi Li, Yulun Wu, Sky Qiu, Yuxuan Li, Mingduo Zhao, Hiroyasu Iso, Mark van der Laan.
ICML 2024
paper

Predicting Cellular Responses with Variational Causal Inference and Refined Relational Information
Yulun Wu, Robert A. Barton, Zichen Wang, Vassilis N. Ioannidis, Carlo De Donno, Layne C. Price, Luis F. Voloch, and George Karypis.
ICLR 2023
paper / code

Spatial Graph Attention and Curiosity-driven Policy for Antiviral Drug Discovery
Yulun Wu, Mikaela Cashman, Nicholas Choma, Erica T. Prates, Verónica G. Melesse Vergara, Manesh Shah, Andrew Chen, Austin Clyde, Thomas S. Brettin, Wibe A. de Jong, Neeraj Kumar, Martha S. Head, Rick L. Stevens, Peter Nugent, Daniel A. Jacobson, and James B. Brown.
ICLR 2022
paper / code

Other Publications

Long-term, non-invasive FTIR detection of low-dose ionizing radiation exposure
Jamie L. Inman*, Yulun Wu*, Liang Chen, Ella Brydon, Dhruba Ghosh, Kenneth H. Wan, Jared De Chant, Lieselotte Obst-Huebl, Kei Nakamura, Corie Y. Ralston, Susan E. Celniker, Jian-Hua Mao, Peter H. Zwart, Hoi-Ying N. Holman, Hang Chang, James B. Brown, and Antoine M. Snijders.
Nature Scientific Reports
paper

Estimating geographic variation of infection fatality ratios during epidemics
Joshua Ladau, Eoin L. Brodie, Nicola Falco, Ishan Bansal, Elijah B. Hoffman, Marcin P. Joachimiak, Ana M. Mora, Angelica M. Walker, Haruko M. Wainwright, Yulun Wu, Mirko Pavicic, Daniel Jacobson, Matthias Hess, James B. Brown, and Katrina Abuabara.
Infectious Disease Modelling
paper

Molecular and functional characterization of the Drosophila melanogaster conserved smORFome
Justin A. Bosch, Nathan Keith, Felipe Escobedo, William W. Fisher, James Thai LaGraff, Jorden Rabasco, Kenneth H. Wan, Richard Weiszmann, Yulun Wu, Yanhui Hu, Shu Kondo, James B. Brown, Norbert Perrimon, and Susan E. Celniker.
Cell Reports
paper