Skip to content
@AI4Science-WestlakeU

AI4Science-WestlakeU

AI for Scientific Simulation and Discovery Lab

Our research group at Westlake University (西湖大学) carries out long-term work on core and universal problems for AI + Science:

  • AI for scientific simulation, design, and control: Developing machine learning algorithms (based on Graph Neural Networks and Diffusion Models) for large-scale, multi-scale scientific simulation (applied to fluid dynamics, materials, plasmas), scientific design (protein design, materials design, mechanical design), and control (fluid control, PDE control);
  • AI for scientific discovery: Developing machine learning algorithms (based on foundation models and neuro-symbolic AI) to discover universal rules and internal structures in scientific systems (applied to life sciences and physics);

Group website: https://ai4s.lab.westlake.edu.cn/

Collaborators (a non-exhaustive list):

Popular repositories Loading

  1. RealPDEBench RealPDEBench Public

    [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data

    Python 126 17

  2. flow_guidance flow_guidance Public

    [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"

    Python 107 10

  3. BuildArena BuildArena Public

    BuildArena, where LLM agents design, build, and test rockets, cars, and bridges in a physics simulator given a goal-directed sentence.

    Python 106 7

  4. wdno wdno Public

    [ICLR 2025] Wavelet Diffusion Neural Operator (WDNO) uses diffusion models on wavelet space for generative PDE simulation and control.

    Python 65 2

  5. scDFM scDFM Public

    scDFM: Distributional Flow Matching for Robust Single-Cell Perturbation Prediction (ICLR 2026)

    Python 58 7

  6. research_toolbox research_toolbox Public

    A useful toolbox for research.

    53 2

Repositories

Showing 10 of 27 repositories
  • group-meeting-materials Public

    Group meeting materials for Tailin Wu's lab

    AI4Science-WestlakeU/group-meeting-materials's past year of commit activity
    12 0 0 0 Updated Sep 15, 2026
  • PACE Public

    PACE, a geometry-aware bridge transport method for recovering continuous cell developmental trajectories from unpaired, destructive single-cell time-course snapshots while respecting the underlying manifold structure.

    AI4Science-WestlakeU/PACE's past year of commit activity
    Python 4 MIT 1 1 0 Updated Jul 26, 2026
  • RealPDEBench Public

    [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data

    AI4Science-WestlakeU/RealPDEBench's past year of commit activity
    Python 126 17 0 0 Updated Jul 16, 2026
  • EqCollide Public

    [KDD26 Oral AI4S Track] The official implementation of the paper "EqCollide: Equivariant and Collision-Aware Deformable Objects Neural Simulator"

    AI4Science-WestlakeU/EqCollide's past year of commit activity
    Python 9 1 0 0 Updated Jun 15, 2026
  • Frontiers-in-Computer-Science-and-Technology-2026 Public

    Frontiers in Computer Science and Technology 2026 Spring

    AI4Science-WestlakeU/Frontiers-in-Computer-Science-and-Technology-2026's past year of commit activity
    Jupyter Notebook 43 MIT 4 0 0 Updated Jun 7, 2026
  • BuildArena Public

    BuildArena, where LLM agents design, build, and test rockets, cars, and bridges in a physics simulator given a goal-directed sentence.

    AI4Science-WestlakeU/BuildArena's past year of commit activity
    Python 106 7 0 0 Updated May 4, 2026
  • ABMS Public

    One step further with Monte-Carlo sampler to guide diffusion better (ICLR 2026)

    AI4Science-WestlakeU/ABMS's past year of commit activity
    Python 8 MIT 0 0 0 Updated Apr 24, 2026
  • scDFM Public

    scDFM: Distributional Flow Matching for Robust Single-Cell Perturbation Prediction (ICLR 2026)

    AI4Science-WestlakeU/scDFM's past year of commit activity
    Python 58 MIT 7 6 0 Updated Apr 22, 2026
  • flow_guidance Public

    [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"

    AI4Science-WestlakeU/flow_guidance's past year of commit activity
    Python 107 MIT 10 1 0 Updated Apr 13, 2026
  • GenCP Public

    Code for reproducing the paper "GenCP: Towards Generative Modeling Paradigm of Coupled physics" (ICLR 2026: https://arxiv.org/abs/2601.19541).

    AI4Science-WestlakeU/GenCP's past year of commit activity
    Python 22 3 0 0 Updated Apr 11, 2026

Top languages

Loading…

Most used topics

Loading…