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

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

    Jupyter Notebook 38 3

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

  3. wdno wdno Public

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

    Python 29

  4. beno beno Public

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

    Python 26 1

  5. t_scend 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.

    Python 23 1

  6. CL_DiffPhyCon CL_DiffPhyCon Public

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

    Python 21 1

Repositories

Showing 10 of 15 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 16 MIT 4 1 0 Updated May 8, 2025
  • FourierFlow Public
    AI4Science-WestlakeU/FourierFlow’s past year of commit activity
    Python 0 MIT 0 0 0 Updated May 4, 2025
  • M2PDE Public
    AI4Science-WestlakeU/M2PDE’s past year of commit activity
    Jupyter Notebook 2 MIT 0 1 0 Updated May 3, 2025
  • wdno Public

    [ICLR2025] 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 29 MIT 0 0 0 Updated Apr 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 21 1 1 0 Updated Apr 19, 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 23 1 0 0 Updated Feb 24, 2025
  • diffphycon Public

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

    AI4Science-WestlakeU/diffphycon’s past year of commit activity
    Jupyter Notebook 38 MIT 3 2 0 Updated Feb 23, 2025
  • safediffcon Public

    Code of Safe Diffusion Models for PDE Control.

    AI4Science-WestlakeU/safediffcon’s past year of commit activity
    3 0 0 0 Updated Feb 4, 2025
  • standard_repo Public template
    AI4Science-WestlakeU/standard_repo’s past year of commit activity
    Jupyter Notebook 5 0 0 0 Updated Jan 18, 2025
  • beno Public

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

    AI4Science-WestlakeU/beno’s past year of commit activity
    Python 26 MIT 1 0 0 Updated Dec 5, 2024

Top languages

Loading…

Most used topics

Loading…