Implicit Neural Spatial Representations for Time-dependent PDEs

Project Page | Paper

Official implementation for the paper:

Implicit Neural Spatial Representations for Time-dependent PDEs
Honglin Chen*¹, Rundi Wu*¹, Eitan Grinspun², Changxi Zheng¹, Peter Yichen Chen^{3 1
1}Columbia University, ²University of Toronto, ³Massachusetts Institute of Technology
ICML 2023

Installation

Prerequisites:

• python 3.9+
• An NVIDIA GPU

Install dependencies with pip:

pip install -r requirements.txt
# NOTE: check https://pytorch.org/ for pytorch installation command for your CUDA version

Experiments

Run each shell script under scripts/ for the examples shown in the paper:

bash scripts/xxx.sh

For instance,

bash scripts/advect1D.sh

Citation

@inproceedings{chenwu2023insr-pde,
    title={Implicit Neural Spatial Representations for Time-dependent PDEs},
    author={Honglin Chen and Rundi Wu and Eitan Grinspun and Changxi Zheng and Peter Yichen Chen},
    booktitle={International Conference on Machine Learning},
    year={2023}
}