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. 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 79 3

  2. flow_guidance flow_guidance Public

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

    Python 65 7

  3. wdno wdno Public

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

    Python 49 2

  4. diffphycon diffphycon Public

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

    Jupyter Notebook 44 4

  5. 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 38 3

  6. beno beno Public

    [ICLR24] A boundary-embedded neural operator that incorporates complex boundary shape and inhomogeneous boundary values

    Python 29 1

Repositories

Showing 10 of 20 repositories
  • 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 0 MIT 0 0 0 Updated Nov 5, 2025
  • diffphycon Public

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

    AI4Science-WestlakeU/diffphycon’s past year of commit activity
    Jupyter Notebook 44 MIT 4 1 0 Updated Nov 4, 2025
  • 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 79 3 1 0 Updated Nov 4, 2025
  • t_scend Public

    This repo is the code for T-SCEND, a novel framework that significantly improves diffusion model’s reasoning capabilities with better energy-based training and scaling up test-time computation.

    AI4Science-WestlakeU/t_scend’s past year of commit activity
    Python 26 MIT 1 0 0 Updated Oct 19, 2025
  • FLDmamba Public
    AI4Science-WestlakeU/FLDmamba’s past year of commit activity
    Python 5 MIT 1 1 0 Updated Oct 20, 2025
  • safediffcon Public

    [ICML2025] SafeDiffCon is a diffusion model for safe PDE Control.

    AI4Science-WestlakeU/safediffcon’s past year of commit activity
    Python 12 MIT 0 0 0 Updated Oct 20, 2025
  • standard_repo Public template
    AI4Science-WestlakeU/standard_repo’s past year of commit activity
    Python 5 MIT 0 0 0 Updated Oct 19, 2025
  • CL_DiffPhyCon Public

    [ICLR 2025] CL-DiffPhyCon achieves closed-loop diffusion control of physical systems with significant speedup of sampling efficiency

    AI4Science-WestlakeU/CL_DiffPhyCon’s past year of commit activity
    Python 27 MIT 2 1 0 Updated Oct 19, 2025
  • 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

    AI4Science-WestlakeU/cindm’s past year of commit activity
    Jupyter Notebook 38 MIT 3 0 0 Updated Oct 3, 2025
  • 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 65 MIT 7 1 1 Updated Sep 16, 2025

Most used topics

Loading…