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. flow_guidance flow_guidance Public

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

    Python 98 9

  2. 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 88 3

  3. RealPDEBench RealPDEBench Public

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

    Python 73 10

  4. wdno wdno Public

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

    Python 60 2

  5. diffphycon diffphycon Public

    [NeurIPS2024] DiffPhyCon uses generative models to control complex physical systems

    Jupyter Notebook 47 4

  6. cindm cindm Public

    [ICLR24] CinDM uses compositional generative models to design boundaries and initial states significantly more complex than the ones seen in training for physical simulation

    Jupyter Notebook 40 3

Repositories

Showing 10 of 25 repositories
  • 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 98 MIT 9 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 15 2 0 0 Updated Apr 11, 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 23 MIT 3 3 0 Updated Apr 7, 2026
  • VFScale Public

    [ICLR26] VFScale: First Verifier-free Test-time Scalable Diffusion Model for Intrinsic Reasoning.

    AI4Science-WestlakeU/VFScale’s past year of commit activity
    Python 5 MIT 0 1 0 Updated Apr 1, 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 32 MIT 3 0 0 Updated Apr 1, 2026
  • M2PDE Public

    [ICML 2025] M2PDE: Compositional Generative Multiphysics and Multi-component PDE Simulation.

    AI4Science-WestlakeU/M2PDE’s past year of commit activity
    Jupyter Notebook 23 1 1 0 Updated Apr 1, 2026
  • wdno Public

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

    AI4Science-WestlakeU/wdno’s past year of commit activity
    Python 60 MIT 2 0 0 Updated Mar 8, 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 73 10 0 0 Updated Mar 8, 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 88 3 0 0 Updated Feb 25, 2026
  • cbi-cfm Public

    Repository for Bidirectional Conditional Flow Matching (Bi-CFM) and Conservation-constrained Bi-CFM (CBi-CFM).

    AI4Science-WestlakeU/cbi-cfm’s past year of commit activity
    Python 1 MIT 0 0 0 Updated Nov 4, 2025

Most used topics

Loading…