Zeyu Xia ☕️

Zeyu Xia

(he/him)

PhD student

University of Virginia

Professional Summary

Zeyu Xia is pursuing his Ph.D. in Computer Science at the University of Virginia, fortunately under the expert guidance of the esteemed Prof. Geoffrey Fox. His research focuses on AI for science, with a particular emphasis on generative models, differentiable simulation and inverse problems. Zeyu is driven by a passion for innovation and a commitment to making cutting-edge technologies accessible to a broader audience.
Download CV

Education

PhD in Computer Science

Aug 2024
May 2029

University of Virginia

Master of Computer Science

Aug 2024
Dec 2025

University of Virginia

Bachelor of Information Technology

Jul 2022
Jun 2023

Queensland University of Technology

MicroMaster in Statistics and Data Science

Feb 2021
May 2023

Massachusetts Institute of Technology on edX

Bachelor of Software Engineering

Jun 2019
Jun 2022

Jinling Institute of Technology

Interests

AI for Science Generative Models Differentiable Simulation Inverse Problems
📚 My Research

I am a PhD student in Prof. Geoffrey Fox’s research group at UVA, with a keen interest in AI for science.

My research focuses on AI for science, with a particular emphasis on generative models, differentiable simulation, and inverse problems.

I would love to collaborate, please feel free to reach out! 😃

Featured Publications
Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universe featured image
Flow Matching

Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universe

Cosmo3DFlow reconstructs the early universe from present-day cosmic structure by combining the 3D Discrete Wavelet Transform with flow matching, compressing sparse 3D volumes into …

Md Khairul Islam
•
Read more
FireDataForge: A Unified Framework for Multi-Source Wildfire Data Retrieval and Integration featured image
Wildfire Data

FireDataForge: A Unified Framework for Multi-Source Wildfire Data Retrieval and Integration

FireDataForge is an open-source Python framework that automates retrieval and harmonization of 11 wildfire-related geospatial data sources, producing analysis-ready arrays for fire …

avatar
Zeyu Xia
•
Read more
ScatterPrism: convergence for generative simulation and inverse problems in particle and nuclear physics featured image
Conditional Flow Matching

ScatterPrism: convergence for generative simulation and inverse problems in particle and nuclear physics

ScatterPrism is a Conditional Flow Matching generative surrogate that replaces slow Monte Carlo simulation of particle collisions and inverts it to infer physics from detector …

avatar
Zeyu Xia
•
Read more
PyTorchFire: A GPU-accelerated wildfire simulator with Differentiable Cellular Automata featured image
Wildfire Simulation

PyTorchFire: A GPU-accelerated wildfire simulator with Differentiable Cellular Automata

PyTorchFire is an open-access, PyTorch-based software utilizing GPU acceleration and a differentiable wildfire Cellular Automata model to achieve rapid, accurate, and realistic …

avatar
Zeyu Xia
•
Read more
Recent Publications
Md Khairul Islam, Zeyu Xia, Ryan Goudjil, Jialu Wang, Arya Farahi, Judy Fox (2026). Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universe. Proceedings of the 32nd ACM SIGKDD Conference on Knowledge Discovery and Data Mining V.2 (KDD 2026).
PDF arXiv GitHub DOI
Zeyu Xia, Lexie Chen, Ye Liu, Huilin Huang (2026). FireDataForge: A Unified Framework for Multi-Source Wildfire Data Retrieval and Integration. 2026 IEEE International Conference on Information Reuse and Integration for Data Science (IEEE IRI 2026).
PDF arXiv GitHub Code Dataset DOI
Zeyu Xia, Tyler Kim, Trevor Reed, Judy Fox, Geoffrey Fox, Adam Szczepaniak (2026). ScatterPrism: convergence for generative simulation and inverse problems in particle and nuclear physics. Journal of Instrumentation, 21(07), C07012.
PDF arXiv GitHub Code Dataset DOI
Zeyu Xia, Sibo Cheng (2025). PyTorchFire: A GPU-accelerated wildfire simulator with Differentiable Cellular Automata. Environmental Modelling & Software, 188, 106401.
PDF arXiv GitHub Code Dataset Video DOI
Zeyu Xia
Equal contribution
, Kan Ma
Equal contribution
, Sibo Cheng, Thomas Blackburn, Ziling Peng, Kewei Zhu, Weihang Zhang, Dunhui Xiao, Alexander J Knowles, Rossella Arcucci (2023). Accurate identification and measurement of the precipitate area by two-stage deep neural networks in novel chromium-based alloys. Physical Chemistry Chemical Physics, 25(23), 15970-15987.
PDF arXiv GitHub Code Dataset DOI
See all publications