Curriculum Vitae

anran.jiao at yale.edu

Updated November 2026

Research Profile

Interests

  • Scientific Machine Learningneural operators, data-efficient modeling, Bayesian inverse problems
  • AI for Scientific Discoveryinterpretable law discovery in complex physical systems
  • Agentic Reasoning & Planningmulti-agent symbolic discovery, procedural memory, tool orchestration

Highlights

First-author publications in Nature Communications and SIAM Journal on Scientific Computing on one-shot operator learning and physics-informed inverse problems (Google Scholar 500+ citations). Current research in agentic scientific discovery of interpretable laws in complex physical systems. Industrial research in agent planning at Amazon, and neural surrogate modeling at FM.

Education

Yale University

Expected May 2027

New Haven, CT

Ph.D. Candidate in Statistics & Data Science

M.A. in Statistics

Dec 2025

GPA 4.0/4.0

University of Pennsylvania

May 2023

Philadelphia, PA

M.S.E. in Data Science

Outstanding Academic Award

Nankai University

Jun 2020

Tianjin, China

B.S. in Applied Physics

Top 5%, Yan Jici Physics Elites Class

National Scholarship

Work Experience

Amazon

May 2026 – Aug 2026

Applied Scientist Intern, Buyer Risk Prevention | Agent Planning & Tool Orchestration

Seattle, WA

  • Developed DAGENT, an end-to-end agentic planning and execution framework for autonomous business metric investigations. Designed a schema-constrained procedural knowledge graph (PKG) to transform past investigations into reusable memory. Implemented multi-case retrieval and objective decomposition with graph-grounded planning to construct dependency-aware execution DAGs for parallel tool orchestration and evidence-guided replanning.
  • PKG-grounded planning achieved 2.17× the accuracy of RAG on SOP-Bench, while DAGENT outperformed the strongest graph-planning baseline by 12.6 percentage points. Integrated into an Amazon deep-dive product.

Factory Mutual

Jun 2024 – Aug 2024

Machine Learning and Fire Modeling Intern

Neural Operators & Radiative Transfer Surrogate

Norwood, MA

  • Developed nested Fourier-MIONet surrogates for the 5D radiative transfer equation in 2D and 3D fire simulations, achieving ∼3% relative L2 error and faster inference than FireFOAM, with generalization across 14–58 kW heat-release rates.
  • Implemented PCA-, KAN-, and Fourier-enhanced MIONets on 2D pool fires and scaled to large 3D McCaffrey fires (∼2.7k grid points).
  • Engineered an HPC-to-cloud workflow spanning HPC clusters, Amazon S3, and AWS SageMaker for efficient data transfer, processing, and training on FireFOAM simulation datasets.

Research Experience

Yale University

Aug 2023 – Present

Graduate Research Assistant

Advisor: Prof. Lu Lu

New Haven, CT

  • One-shot neural operator learning

    Proposed a data-efficient method for PDE solution operators that learns from only one data pair by leveraging the locality of derivatives, enabling generalization to varied new inputs and geometries.

  • Symmetry-constrained scientific law discovery

    Developed a multi-agent reinforcement learning framework with structure-aware LSTMs and physical symmetry constraints to discover interpretable constitutive laws in crystal plasticity across 120 crystal orientations and 12 interacting slip systems. Identified nonlinear pairwise coupling beyond conventional additive hardening, reducing shear-rate relative L2 error from 18.1% to 9.7%.

  • Physics-informed neural operators on unknown manifolds

    Developed physics-informed DeepONet for PDEs on unknown manifolds, using RBF kernels and diffusion maps to approximate differential operators. Integrated the method into a Bayesian MCMC framework to estimate diffusion coefficients from sparse, noisy point-cloud observations.

  • Physics-informed ML for chromatin inference

    Developed a parallel physics-informed neural network using mixed formulation and domain decomposition to infer spatially varying biophysical parameters (methylation rates and chromatin–lamina affinity) from single-cell STORM microscopy data, demonstrating robust recovery under up to 5% observational noise in synthetic benchmarks.

  • Time-adaptive neural operator learning

    Developed TANTE, a time-adaptive Transformer-based operator-learning framework using neural Taylor expansion for continuous-time PDE prediction, achieving 60–80% performance improvements and 30–40% inference speedups over baselines.

University of Pennsylvania

Jan 2022 – May 2023

Research Assistant

Lu group, Scientific Machine Learning

Philadelphia, PA

  • Out-of-distribution operator adaptation

    Implemented DeepONet fine-tuning with new observations to improve out-of-distribution generalization of learned PDE solution operators.

  • Open-source scientific ML

    Contributed to DeepXDE, an open-source library for scientific ML and AI for PDEs. Implemented physics-informed neural networks (PINNs) to solve complex inverse problems and parameter estimation tasks in heterogeneous domains, including photonic nanostructures, quantitative finance, and chromatin organization.

Selected Publications

Google Scholar

  1. A. Jiao, H. He, R. Ranade, J. Pathak, & L. Lu. One-shot learning for solution operators of partial differential equations. Nature Communications, 16(1):8386, 2025.
  2. A. Jiao, Q. Yan, J. Harlim, & L. Lu. Solving forward and inverse partial differential equation problems on unknown manifolds via physics-informed neural operators. SIAM Journal on Scientific Computing, 48(1):C136–C163, 2026.
  3. A. Jiao, W. Jiang, X. Lu, Y. Wang, & L. Lu. Nested Fourier-enhanced neural operator for efficient modeling of radiation transfer in fires. arXiv:2604.13919, 2026.
  4. Z. Wu, S. Wang, S. Zhang, S. He, M. Zhu, A. Jiao, L. Lu, & D. van Dijk. TANTE: Time-adaptive operator learning via neural Taylor expansion. Journal of Computational Physics, 562:115041, 2026.
  5. P. Xiao, M. Zheng, A. Jiao, X. Yang, & L. Lu. Quantum DeepONet: Neural operators accelerated by quantum computing. Quantum, 9:1761, 2025.
  6. M. Zhu, H. Zhang, A. Jiao, G. Karniadakis, & L. Lu. Reliable extrapolation of deep neural operators informed by physics or sparse observations. Computer Methods in Applied Mechanics and Engineering, 412:116064, 2023.
  7. B. Fan, E. Qiao, A. Jiao, Z. Gu, W. Li, & L. Lu. Deep learning for solving and estimating dynamic macro-finance models. Computational Economics, 65(6):3885–3921, 2025.

Talks & Presentations

  • Symmetry-constrained discovery of crystal plasticity laws from DDD, SIAM Conference on Mathematics of Data Science, 2026.
  • Guest lecture for the course Data-driven approaches for partial differential equations: Fundamentals and selected topics, University of Stuttgart, 2025.
  • Nested Fourier-MIONet for radiation transfer in fires, FM Fire Modeling Workshop, 2025.
  • From physics-informed machine learning to one-shot operator learning for PDEs, Foundation Models of the Physical Universe Symposium, Foresite Labs, 2025.
  • One-shot learning for solution operators of partial differential equations, North American High Order Methods Conference, 2024.
  • One-shot learning for solution operators of partial differential equations, 1st Annual Meeting of the New York-New Jersey-Pennsylvania Section of SIAM, 2023.

Professional Service

Reviewing

ICLR, NeurIPS, IEEE TNNLS, Journal of Machine Learning, Ain Shams Engineering Journal.

Teaching

Teaching Fellow

  • Data Analysis 2025
  • Introductory Machine Learning 2024
  • Artificial Intelligence 2022

Technical Skills

Programming & ML: Python, Java, MATLAB, SQL, R, PyTorch, JAX, TensorFlow, LLM-based Agents.

Tools & Computing: Git, Linux, HPC/Slurm, AWS S3, SageMaker.