Physics-aware Recurrent Convolutional Neural Networks (PARCv2) in PyTorch
Paper link: arxiv ICML 2024 Proceedings
Install via miniforge (open source version of conda):
mamba create -n parc python=3.12
mamba activate parc
mamba install pytorch pytorch-cuda=12.4 -c pytorch -c nvidia
mamba install numpy matplotlib tqdm ipykernel imageio
pip install -e .If you find our research helpful, please consider citing us with：
@InProceedings{pmlr-v235-nguyen24c,
title = {{PARC}v2: Physics-aware Recurrent Convolutional Neural Networks for Spatiotemporal Dynamics Modeling},
author = {Nguyen, Phong C.H. and Cheng, Xinlun and Azarfar, Shahab and Seshadri, Pradeep and Nguyen, Yen T. and Kim, Munho and Choi, Sanghun and Udaykumar, H.S. and Baek, Stephen},
booktitle = {Proceedings of the 41st International Conference on Machine Learning},
pages = {37649--37666},
year = {2024},
editor = {Salakhutdinov, Ruslan and Kolter, Zico and Heller, Katherine and Weller, Adrian and Oliver, Nuria and Scarlett, Jonathan and Berkenkamp, Felix},
volume = {235},
series = {Proceedings of Machine Learning Research},
month = {21--27 Jul},
publisher = {PMLR},
pdf = {https://raw.githubusercontent.com/mlresearch/v235/main/assets/nguyen24c/nguyen24c.pdf},
url = {https://proceedings.mlr.press/v235/nguyen24c.html}
}