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 103 10

  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 90 3

  3. RealPDEBench RealPDEBench Public

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

    Python 78 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

Top languages

Loading…

Most used topics

Loading…